基于条件约束下的自注意力生成对抗网络的图像修复

摘要

图像修复是图像处理领域的重要研究方向.为了解决现阶段图像修复算法存在修复区域与周围区域不一致的模糊纹理问题，提出了一种图像修复方法.该方法在生成对抗网络的基础上，引入条件特征和自注意模块，并将数据的具体维度与语义特征相关联.采用该方法训练的修复模型可以对特定类型的图像进行修复，并且保证了整体修复的一致性和局部信息细节的合理性.实验在CeleBA人脸数据集上进行训练测试，获得了良好的修复结果.

关键词

图像修复 / 自注意力 / 生成对抗网络 / 卷积神经网络