基于注意力机制的残差单幅图像去雨方法

摘要

本发明公开了一种基于注意力机制的残差单幅图像去雨方法，主要解决现有单幅图像去雨技术存在局限性、处理效果不理想的问题。其方案包括：1)对输入图像进行预处理，得到预处理后图像；2)构建包括残差网络模块和编解码器网络模块的注意力残差神经网络模型；3)将预处理后图像输入注意力残差神经网络模型进行训练，利用损失函数对注意力残差神经网络模型进行约束，然后反向传播进行参数更新，得到训练好的去雨神经网络模型；4)将待处理的有雨图像输入去雨神经网络模型中进行图像处理，得到无雨清晰图像。本发明能够有效去除单幅含雨图像中的雨纹，得到清晰图像；同时，对于原图中的背景信息实现充分保留。