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

摘要：

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

关键词：全局特征;通道注意力模块;WGAN-GP;图像彩色化

所属期刊栏目：36

分类号：TP391.4(计算技术、计算机技术)

资助基金：国家重点研发计划No.2018YFC0308100

在线出版日期：2021-10-25（万方平台首次上网日期，不代表论文的发表时间）

页数：共9页

页码：1305-1313