Grayscale Images Colorization with Convolutional Neural Networks

Jiancheng An,Koffi Gagnon Kpeyiton,Qingnan Shi
DOI: https://doi.org/10.1007/s00500-020-04711-3
2020-01-01
Abstract:Previous approaches to the colorization of grayscale images rely on human manual labor and often produce desaturated results that are not likely to be true colorizations. Inspired by [15], we proposed an automatic approach based on deep neural networks to color the image in grayscale. We have studied several models, approaches and loss function to understand the best practices to produce a plausible colorization. We trained a convolutional neural network by noting that some loss functions work better than others. We used the VGG-16 CNN model based on the classification with the loss of cross entropy that is very well to produce a plausible colorization.
What problem does this paper attempt to address?