一种基于注意力跨层转移机制的生成式图像修复方法

摘要

本发明公开了一种基于注意力跨层转移机制的生成式图像修复方法，属于图像修复技术领域，包括如下步骤：将缺损图像和掩码一起输入网络，重构ATNAL特征图，通过跳跃连接将各级ATNAL特征图传递至多尺度解码器对应的解码层；本发明中，通过提出注意力跨层转移网络，在生成器对缺损图像编码完成后由深入浅逐层重构编码特征图，为解码工作提供指导，提出多尺度解码器，使用跳跃连接将ATNAL重构特征图连接至多尺度解码器对应解码层，与上一级潜在特征融合进行解码，最大限度减少解码过程中上下文信息丢失，创新应用L1损失，将解码过程中的各尺度特征图转为RGB图像后使用L1损失进行约束，促使多尺度解码器生成上下文语义一致的精细内容。