结合全局语义优化的对抗性灰度图像彩色化

• 作者: 万园园 王雨青 张晓宁 李荅群 陈小林
• 来源期刊: 液晶与显示
• 关键词: 全局特征 通道注意力模块 WGAN-GP 图像彩色化
• 年，卷（期）: 2021,（9）
• 所属期刊栏目: 图像处理|Image Processing
• 页码范围: 1305-1313
• 页数: 9页
• 分类号: TP391.4
• 语种: 中文
• DOI: 10.37188/CJLCD.2021-0012

摘要

针对当前灰度图像彩色化算法容易出现色彩枯燥、颜色溢出和图像细节损失等问题,本文提出一种结合全局语义优化的对抗性灰度图像彩色化算法.其中,生成网络采用自主改进的U-Net网络.一方面,改进的U Net网络利用多层卷积对输入图像进行逐步下采样,在获取多尺度层级特征和全局特征的同时,在跳跃连接中将全局特征和多尺度层级特征进行自适应融合,从而有效增强算法对全局语义信息的理解能力并缓解颜色溢出的现象;另一方面,改进的U-Net网络在上采样过程中融合通道注意力模块,使得在提取卷积特征时能够有效抑制噪声并降低特征冗余性.判别网络主要采用全卷积结构,通过反向传播误差以达到优化生成网络的目的 .此外,本文算法的损失函数将WGAN-GP网络的优化思想和颜色损失相结合,从而解决传统生成对抗网络训练时出现的梯度消失和模式崩溃等问题.本文算法在Place365测试集上所获取的峰值信噪比、结构相似性和信息熵指标分别为24.455 dB、0.943和7.489.实验结果表明,本文算法能够缓解颜色溢出,且细节保持和颜色饱和度方面都具有一定优势.