一种基于可变形注意力机制的图像去雨方法

本发明涉及计算机视觉技术领域，具体涉及一种基于可变形注意力机制的图像去雨方法，包括以下步骤：构建雨水图像和干净图像对数据集；生成模块基于可变形注意力机制对雨水图像进行处理，生成伪去雨图像；判别模块对所述伪去雨图像和干净图像进行判断，获取判别结果；基于所述判别结果，调整所述生成模块和所述判别模块的参数，对图像去雨模型进行训练，使得所述生成模块和所述判别模块性能达到最优；将训练完成的图像去雨模型应用于图像雨水去噪，获取去雨图像。本发明以生成对抗网络为基础，引入了可变形注意力机制，从多尺度层面适应不同细粒度的雨水噪点，能够提升对雨水噪点的去除能力。