Automatic colorization using fully convolutional networks

作者：Zhuge, Jingjing; Lin, Jiajun^*; An, Wei

来源：Journal of Electronic Imaging, 2018, 27(4): 043025.

DOI：10.1117/1.JEI.27.4.043025

• colorization
• fully convolutional networks
• neural network
• level-set method

We propose an approach for automatically colorizing grayscale images using fully convolutional networks (FCNs). In contrast to traditional colorization methods, our approach operates only on grayscale images without any manual assistance. We first build an end-to-end deep learning network based on an FCN. Global, midlevel, and local features are extracted from the network and fused to construct each deconvolutional layer. To ensure color consistency, a low-frequency regularization term is presented to maintain the coherence between neighboring pixels. We then present an improved level-set method, which we apply to the output of the FCN to repair color bleeding caused by the rough segmentation performed by the FCN. To evaluate our approach, we compare the objective image quality resulting from our method with the results of other methods by assessing the peak signal-to-noise ratio, the mean squared error, the structural similarity index (SSIM), and the multiscale SSIM (MS-SSIM). In addition, we design a Turing test to evaluate the subjective image quality. The results show that our colorized images more closely resemble the ground-truth images and are more robust than those produced via other methods.

• 出版日期2018-7
• 单位华东理工大学; 中国科学院信息工程研究所