摘要
针对基于生成对抗网络的图像修复模型,在修复随机不规则大面积缺失区域的图像时仍存在修复图像出现伪影或失真的问题,提出一种融合卷积块注意力机制(convolutional block attention module,CBAM)的区域归一化图像修复模型。基于生成对抗网络(GAN)的思想,分为生成网络和对抗网络,在修复模型生成网络编码器与解码器之间的残差块中嵌入CBAM注意力机制,组合成包含混合扩张卷积、可学习区域归一化(RN-L)和CBAM注意力机制的残差块,来获得更有效的感受野,并自动检测图像中潜在的缺失和未缺失区域,同时捕获图像缺失区域的关键局部细节信息,提升网络的特征学习能力,进而提取待修复图像中的有效特征;将原来残差块和解码器的激活函数由ReLU改为Mish及ELU,有效减少信息的保留,从而加快网络的训练。在CelebA、Paris StreetView以及Places2数据集上对具有不规则缺失区域的图像进行实验,采用性能评价指标峰值信噪比(PSNR)、结构相似性(SSIM)和L1损失进行评价。实验结果表明,模型的定性和定量分析结果均优于对比模型,其性能评价指标值更高,修复图像的视觉效果更好。模型在修复随机不规则大面积缺失区域的图像时,能有效消除伪影,呈现出更好的视觉效果,并且模型的收敛速率更快。
作者
李筱玉
张乾
周遵富
王轩云
何兴
LI Xiaoyu;ZHANG Qian;ZHOU Zunfu;WANG Xuanyun;HE Xing
出处
《信息技术与信息化》
2023年第10期136-143,共8页
Information Technology and Informatization
基金
贵州民族大学校级科研项目(GZMUZK[2021]YB23)
黔教合YJSKYJJ[2021]121。