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

技术特征：
1.一种基于注意力跨层转移机制的生成式图像修复方法，包括如下步骤：s1、将缺损图像和掩码一起输入网络；s2、重构atnal特征图；s3、通过跳跃连接将各级atnal特征图传递至多尺度解码器对应的解码层；s4、输出下一层解码特征图；s5、通过上采样得到输出图像；s6、在解码完成后将各级输出特征图转为rgb图像；s7、区分真实图像与修复图像；s8、对修复图像进行评判；s9、提升图像修复质量。2.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s1中，将缺损图像和掩码一起输入网络，进行编码。3.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s2中，atnal编码器在编码完成后，由深到浅逐层重构atnal特征图。4.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s3中，通过跳跃连接将各级atnal特征图传递至多尺度解码器对应的解码层，其中最深层解码特征图由对应的编码特征图反卷积而来。5.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s4中，与对应尺度的atnal特征图融合后进行解码输出下一层解码特征图。6.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s5中，通过上采样得到尺寸大小为256*256的输出图像。7.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s6中，在解码完成后将多尺度解码器各级输出特征图转为rgb图像。8.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s7中，使用对抗损失对图像判别器进行约束，区分真实图像与修复图像。9.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s8中，引用风格损失和感知损失对修复图像进行评判。10.根据权利要求1所述的一种基于注意力跨层转移机制的生成式图像修复方法，其特征在于，所述s9中，最小化各级解码层最终输出的rgb图像与真实图像之间的l1损失对生成器进行约束，提升图像修复质量。

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


技术研发人员：樊瑶 石英男
受保护的技术使用者：樊瑶
技术研发日：2021.12.24
技术公布日：2022/4/1