基于自交叉融合注意力机制的图像修复方法及系统

本发明属于计算机视觉图像修复领域，提供了一种基于自交叉融合注意力机制的图像修复方法及系统，包括获取高质量图重建训练集以及低质量图修复训练集；第一阶段是基于预训练的编码器、解码器、交叉注意力模块以及代码本，利用高质量图重建训练集，对交叉注意力模型进行参数优化；第二阶段是基于预训练的编码器、解码器、自交叉融合注意力模块以及代码本，利用低质量图训练集，对编码器和自交叉融合注意力模块进行参数优化；经过第一阶段和第二阶段的训练，得到训练好的图像修复模型，利用训练好的图像修复模型对破损图像进行修复。本发明提出自交叉融合注意力机制替换离散的向量量化过程，在保证修复图质量的同时，保证修复图的真实性和可靠性。