一种改进的生成对抗网络的图像上色方法研究

作者：蒋文杰; 罗晓曙^*; 戴沁璇

来源：计算机技术与发展, 2020, 30(07): 56-59.

• 黑白照片/图像
• 上色
• 卷积神经网络
• 无监督学习
• 生成对抗网络

利用手持云台对黑白照/图像进行拍摄再对其进行上色是耗时耗力的工作,为了提高对黑白照片/图像上色的效率和视觉效果,利用深度学习的卷积神经网络提出了一种基于改进生成对抗网络的上色算法。采用了原模型pix2pix(image-to-image translation with conditional generative adversarial networks)的U型结构的生成器并在其中引入了自注意力机制提高输出图像的色彩多样性;其次是使用实例归一化对网络结构作进一步的优化处理,生成器和判别器在网络训练期间不断进行相互对抗学习,同时模型不断学习并优化黑白照片/图像到对应彩色图像的映射关系;最后实现了对黑白照片/图像的自动化上色,同时使用客观量化指标(MSE、PSNR和SSIM)对实验结果进行评价。实验结果表明:该算法能快速有效实现对黑白照片/图像的无监督上色,同时保持良好的视觉效果。

• 单位
  广西师范大学; 电子工程学院