计算机工程与应用 ›› 2020, Vol. 56 ›› Issue (16): 183-190.DOI: 10.3778/j.issn.1002-8331.1907-0222

• 图形图像处理 • 上一篇    下一篇

基于联合一致循环生成对抗网络的人像着色

刘昌通,曹林,杜康宁   

  1. 1.北京信息科技大学 光电测试技术及仪器教育部重点实验室,北京 100101
    2.北京信息科技大学 信息与通信工程学院,北京 100101
  • 出版日期:2020-08-15 发布日期:2020-08-11

Portrait Coloring Based on Joint Consistent Cyclic Generative Adversarial Network

LIU Changtong, CAO Lin, DU Kangning   

  1. 1.Key Laboratory of the Ministry of Education for Optoelectronic Measurement Technology and Instrument, Beijing Information Science and Technology University, Beijing 100101, China
    2.School of Information and Communication Engineering, Beijing Information Science and Technology University, Beijing 100101, China
  • Online:2020-08-15 Published:2020-08-11

摘要:

传统灰度图像着色方法存在颜色失真、效果不佳等问题,已逐渐被深度学习方法取代。目前基于深度学习的人像着色方法主要存在复杂背景下误着色的问题。针对上述问题,提出了联合一致循环生成对抗网络的人像着色方法。该方法在循环生成对抗网络的基础上,采用联合的一致性损失训练模型;生成网络采用U型网络结构(UNet)进行改进,以提高生成图像信息的完整性;判别网络中引入多特征融合的特征提取方式,增强特征对图像的细节表达。最后通过在自建的CASIA-PlusColors高质量人像数据集中的对比实验,验证了该方法对复杂背景中的人像着色有着更好的效果。

关键词: 人像着色, 联合一致循环生成对抗网络, 深度学习, 特征融合

Abstract:

Traditional grayscale image coloring methods have problems such as color distortion and poor effect, and have been gradually replaced by deep learning methods. At present, the portrait coloring method based on deep learning mainly has the problem of portrait miscoloring in complex background. In view of the above problems, this paper proposes a portrait coloring method for joint consistent cyclic generative adversarial network. The method uses a joint consistency loss training model based on the cycle generative adversarial network. The generative network uses the U-network structure (UNet) to improve the model generative network and enhance the integrity of the generated image information. The feature extraction method of multi-feature fusion is introduced in the adversarial network, and the feature expression of the feature is enhanced. Finally, through the comparison experiments in the self-built CASIA-PlusColors high-quality portrait dataset, it is verified that the proposed method has a better effect on portrait coloring in complex background.

Key words: portrait coloring, joint consistent cycle generative adversarial network, deep learning, feature fusion