基于密集卷积生成对抗网络的图像修复

摘要：

图像修复是一项利用缺损图像中已知信息对缺损区域信息进行估计修复的技术.针对大面积语义信息缺失的图像进行修复时,若训练数据集较小且图像背景相对复杂,则基于生成模型的修复结果常出现模糊、伪影和视觉相似度差等问题.针对上述问题,文中提出了一种基于密集卷积生成对抗网络的图像修复算法.该算法采用生成对抗网络作为图像修复的基本框架.首先,利用密集卷积块构建具有编解码结构的生成网络,不但加强了图像特征的提取,提高了图像修复能力,而且避免了深度增加引起的梯度消失问题.其次,在编码和解码结构之间引入跳跃连接,解决了网络层间信息传递丢失的问题.然后,在网络优化过程中,结合重建损失、对抗损失和TV损失来训练网络模型,增强了网络稳定性.最后,分别在CelebA和Car两个数据集上进行实验,所提算法的修复结果在视觉效果、峰值信噪比PSNR和结构相似度SSIM 3个方面均优于3种代表性图像修复算法,其有效性得到验证.

关键词：生成对抗网络、图像修复、密集卷积块、跳跃连接、损失函数

所属期刊栏目：47

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

资助基金：国家自然科学基金;北京市自然科学基金;北京市教委科学研究计划

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

页数：共6页

页码：202-207