Implementation of image colorization with convolutional neural network

Abstract

Huge amount of work is getting done on Image colorization worldwide. This research paper proposes a model for image colorization while making use of fully automatic Convolutional Neural Network. Image colorization processes a daunting task, and this research paper proposes a relevant model for the prediction of A and B models for LAB color space and it makes a direct use the lightness channel. In this work, a pre-trained VGG-16 network was used for semantically interpreting the concepts associated with images and coloring the images. In the proposed work, the convolutional layer has been fused with the max pooling layer (higher one) of the VGG network. Architecture of the proposed model has been presented. The experimentation has been carried out with varying objective functions. LaMem experimental dataset has been used in this work in order to validate the proposed model. The proposed model is evaluated and results are visualized by histograms for true and predicted images for RGB values. Further, the proposed model has been compared with the existing models and performs better in terms of execution times (in s) for different image sizes and the results are tabulated.