一种基于雨纹优化的全局注意力单幅图像去雨方法

本发明公开了一种基于雨纹优化的全局注意力单幅图像去雨方法，涉及图像处理领域。包括：获取待去雨图像及其对应的真实无雨图像作为训练样本集；利用训练样本集训练去雨网络，去雨网络包括雨纹先验特征提取模块和基于上下文特征融合的U型编‑解码器，由雨纹先验特征提取模块获取待去雨图像的雨纹先验特征，将雨纹先验特征与重叠编码后的待去雨图像特征作为基于上下文特征融合的U型编‑解码器的输入，将解码结果恢复通道数，与待去雨图像相加后生成去雨图像；利用训练后的去雨网络完成图像去雨任务。本发明结合雨纹先验知识来保留全局注意力长距离建模所忽略的雨纹结构和背景细节特征，拟合全局不规则雨纹的同时增强了对背景细节的表征能力。