一种大坝表面图像修复方法、装置、设备及存储介质

本发明涉及图像识别技术领域，公开了一种大坝表面图像修复方法、装置、设备及存储介质，本发明通过在第一生成对抗网络中添加自注意力机制模块得到第二对抗生成网络，进一步，将自注意力机制模块的输出作为生成器的输入并与判别器进行迭代对抗训练直至生成优化后的目标对抗生成网络，解决了生成对抗网络难以收敛、不稳定的问题。最后，通过优化后的目标生成对抗网络进行大坝表面图像的修复，复原了大坝表面被杂物遮挡的部分，生成了完整的高质量的大坝表面图像，为大坝的维护提供了支持。