Options
Notice of Removal: Robust image colorization using self attention based progressive generative adversarial network
ISSN
21607508
Date Issued
2019-06-01
Author(s)
Sharma, Manoj
Makwana, Megh
Upadhyay, Avinash
Singh, Ajay Pratap
Badhwar, Anuj
Trivedi, Akkshita
Saini, Anil
Chaudhury, Santanu
DOI
10.1109/CVPRW.2019.00272
Abstract
Automatic image colorization is a very interesting computer graphics problem wherein an input grayscale image is transformed into its RGB domain. However, it is an ill-posed problem as there can be multiple RGB outcomes for a particular grayscale pixel. The problem further complicates if noise is present in the grayscale image. In this paper, we propose a Robust Image Colorization using Self-attention based Progressive Generative Adversarial Network (RIC-SPGAN) which consists of residual encoder-decoder (RED) network and a Self-attention based progressive Generative network (SP-GAN) in a cascaded form to perform the denoising and colorization of the image. We have used self-attention based progressive network to model the long-range dependencies and gradually enhanced the resolution of the colorized image for faster, stable and variation rich features for generation of the image. We also presented the stabilization technique of the presented generative model. Our model has shown exceptional perceptual results on noisy and normal grayscale images. We have trained our model on ILSVRC2012. The visual results on images of DIV2K with and without noise has been presented in the paper along with the failure cases of the model.