一种结合小波卷积和Transformer的两阶段图像修复方法

本发明涉及一种结合小波卷积和Transformer的两阶段图像修复方法，属于计算机视觉领域，用于在图像修复任务中有效恢复丢失区域的细节信息。该方法包括以下两个阶段：第一阶段，采用小波卷积作为编码器和解码器，并在中间部分引入残差结构，构建边缘修复网络。通过小波卷积对图像进行多尺度分解，有效提取图像的细节特征，并利用残差结构增强特征的表达能力，从而生成精确的边缘信息；第二阶段将第一阶段生成的边缘信息与缺失图像结合，作为输入传递至图像修复网络，通过自注意力机制进行全局上下文建模，实现高质量的图像修复。通过该方法，能够提高图像修复的精度和质量，同时在不依赖于复杂的先验信息的情况下，保持图像的结构一致性与视觉自然性。