DHAN-WGAN: Adversarial Image Colorization Using Deep Hybrid Attention Aggregation Network

He Feng,Yingyun Yang
DOI: https://doi.org/10.1109/cost60524.2023.00041
2023-01-01
Abstract:While the gray-scale image colorization algorithms based on deep learning bring automatic and efficient coloring methods, there are still a series of problems such as inaccurate coloring details, great ill-posed coloring effects, and low saturation of the generated images. To address these issues, this paper proposes a Deep Hybrid Attention Aggregation Network based on Wasserstein Generative Adversarial Network: DHAN-WGAN. In the generator, we use dynamic convolution and attention mechanisms for finer coloring, and a global prior of image scene categories is incorporated to achieve better colorization. In view of the deficiencies of traditional Generative Adversarial Networks (GANs), this paper uses improved Wasserstein GAN with Gradient Penalty (WGAN-GP) for more stable training. Experiments prove that DHAN-WGAN can have a more accurate and more saturated colorization effect on gray-scale images.
What problem does this paper attempt to address?