基于数据增强和频率Mamba的数字壁画修复方法

一种基于频率信息和Mamba的数字壁画修复方法，由构建壁画修复网络模型训练数据集，并扩增模型训练数据集、构建壁画修复网络模型、训练壁画修复网络模型和利用训练好的壁画修复网络模型对破损壁画进行修复四部分组成。本发明利用随机游走方法生成不规则的掩膜图像，以实现对受损壁画类型的模拟，使用壁画数据增强方法扩充模型的训练数据集，来增强模型的泛化性，壁画修复网络模型包含了第一生成网络、第二生成网络和判别网络，在第一、第二生成网络的编码器、特征变换网络和解码器中捕获了频域信息，同时特征变换网络中还捕获了长序列的依赖关系，提升了修复质量。本发明对含有不规则形状破损的壁画，具有修复结果自然、美观，纹理清晰等优点。