[发明专利]一种引入注意力机制的人脸图像修复方法

摘要

本发明涉及一种引入注意力机制的人脸图像修复方法，该方法包括：(1)获取原始数据集并进行图像预处理，得到我们需要的人脸图像数据集并合理划分整理为测试集与数据集。(2)将训练数据集输入到引入上下文关注层的图像修复模型中进行训练，该模型在生成器网络中引入了两个并行的编码器网络，一个编码器网络用于进行卷积操作提取高级特征图像，另一个编码器用于引入上下文关注层网络，用于实现前景区域与背景区域之间的长程关联。(3)将测试数据集输入到训练好的人脸修复模型中，测试训练好的修复模型对于缺损人脸图像的修复能力。本发明中引入上下文关注层后，解决了卷积神经网络由于感受野大小有限导致修复模型无法充分利用背景区域信息的问题，实现了背景信息与前景区域的长程关联，充分利用了背景区域信息对前景区域进行填充。引入上下文关注层后，修复模型在一些细节纹理上取得了更好的修复效果，总体上也提升了人脸图像的修复效果。