An Improved Encoder-Decoder CNN with Region-Based Filtering for Vibrant Colorization  

作　　者：Mrityunjoy Gain Md Arifur Rahman Rameswar Debnath MrimMAlnfiai Abdullah Sheikh Mehedi Masud Anupam Kumar Bairagi 

机构地区：[1]Computer Science and Engineering Discipline,Khulna University,Khulna 9208,Bangladesh [2]Department of Information Technology,College of Computers and Information Technology,Taif University,Taif 21944,Saudi Arabia [3]Department of Computer Science,College of Computers and Information Technology,Taif University,Taif 21944,Saudi Arabia

出　　处：《Computer Systems Science & Engineering》2023年第7期1059-1077,共19页计算机系统科学与工程（英文）

基　　金：Taif University Researchers Supporting Project Number(TURSP-2020/10),Taif University,Taif,Saudi Arabia.

摘　　要：Colorization is the practice of adding appropriate chromatic values to monochrome photographs or videos.A real-valued luminance image can be mapped to a three-dimensional color image.However,it is a severely ill-defined problem and not has a single solution.In this paper,an encoder-decoder Convolutional Neural Network(CNN)model is used for colorizing gray images where the encoder is a Densely Connected Convolutional Network(DenseNet)and the decoder is a conventional CNN.The DenseNet extracts image features from gray images and the conventional CNN outputs a^(*)b^(*)color channels.Due to a large number of desaturated color components compared to saturated color components in the training images,the saturated color components have a strong tendency towards desaturated color components in the predicted a^(*)b^(*)channel.To solve the problems,we rebalance the predicted a^(*)b^(*)color channel by smoothing every subregion individually using the average filter.2 stage k-means clustering technique is applied to divide the subregions.Then we apply Gamma transformation in the entire a^(*)b^(*)channel to saturate the image.We compare our proposed method with several existing methods.From the experimental results,we see that our proposed method has made some notable improvements over the existing methods and color representation of gray-scale images by our proposed method is more plausible to visualize.Additionally,our suggested approach beats other approaches in terms of Peak Signal-to-Noise Ratio(PSNR),Structural Similarity Index Measure(SSIM)and Histogram.

关 键 词：COLORIZATION DenseNet desaturation K-MEANS 

分 类 号：TP393.08[自动化与计算机技术—计算机应用技术]