计算机应用 ›› 2019, Vol. 39 ›› Issue (6): 1816-1823.DOI: 10.11772/j.issn.1001-9081.2018102100

• 虚拟现实与多媒体计算 • 上一篇    下一篇

基于密集神经网络的灰度图像着色算法

张娜, 秦品乐, 曾建潮, 李启   

  1. 中北大学 大数据学院, 太原 030051
  • 收稿日期:2018-10-18 修回日期:2018-12-18 出版日期:2019-06-10 发布日期:2019-06-17
  • 通讯作者: 秦品乐
  • 作者简介:张娜(1995-),女,山西临汾人,硕士研究生,主要研究方向:机器学习、计算机视觉、数字图像处理;秦品乐(1978-),男,山西太原人,副教授,博士,CCF会员,主要研究方向:大数据、机器视觉、三维重建;曾建潮(1963-),男,山西太原人,教授,博士生导师,博士,主要研究方向:复杂系统的维护决策和健康管理;李启(1991-),男,山西大同人,硕士研究生,主要研究方向:机器学习、计算机视觉、数字图像处理。

Grayscale image colorization algorithm based on dense neural network

ZHANG Na, QIN Pinle, ZENG Jianchao, LI Qi   

  1. School of Data Science And Technology, North University of China, Taiyuan Shanxi 030051, China
  • Received:2018-10-18 Revised:2018-12-18 Online:2019-06-10 Published:2019-06-17

摘要: 针对在灰度图像着色领域中,传统算法信息提取率不高、着色效果不理想的问题,提出了基于密集神经网络的灰度图像着色算法,以实现改善着色效果,让人眼更好地观察图片信息的目的。利用密集神经网络的信息提取高效性,构建并训练了一个端到端的深度学习模型,对图像中的各类信息及特征进行提取。训练网络时与原图像进行对比,以逐渐减小网络输出结果的信息、分类等各类型的损失。训练完成后,只需向网络输入一张灰度图片,即可生成一张颜色饱满、鲜明逼真的彩色图片。实验结果表明,引入密集网络后,可有效改善着色过程中的漏色、细节信息损失、对比度低等问题,所提算法着色效果较基于VGG网络及U-Net、双流网络结构、残差网络(ResNet)等性能优异的先进着色算法而言取得了显著的改进。

关键词: 图像着色, 密集神经网络, 灰度图像, 特征利用, 信息损失

Abstract: Aiming at the problem of low information extraction rate of traditional methods and the unideal coloring effect in the grayscale image colorization field, a grayscale image colorization algorithm based on dense neural network was proposed to improve the colorization effect and make the information of image be better observed by human eyes. With making full use of the high information extraction efficiency of dense neural network, an end-to-end deep learning model was built and trained to extract multiple types of information and features in the image. During the training, the loss of the network output result (such as information loss and classification loss) was gradually reduced by comparing with the original image. After the training, with only a grayscale image input into the trained network, a full and vibrant vivid color image was able to be obtained. The experimental results show that the introduction of dense network can effectively alleviate the problems such as color leakage, loss of detail information and low contrast, during the colorization process. The coloring effect has achieved significant improvement compared with the current advanced coloring methods based on Visual Geometry Group (VGG)-net, U-Net, dual stream network structure, Residual Network (ResNet), etc.

Key words: image coloring, dense neural network, grayscale image, feature utilization, information loss

中图分类号: