科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于图像先验和门控注意力学习的单图去雨方法及系统,该方法包括：构建基于图像先验和门控注意力学习的网络模型,所述网络模型包括顺序连接的特征提取器、编解码器和具有注意力机制的门控器；获取训练样本；构成有雨-无雨图像对训练数据集；将所述有雨-无雨图像对训练数据集进行预处理,将预处理后的数据输入到所述基于图像先验和门控注意力学习的网络模型中进行训练,得到训练好的基于图像先验和门控注意力学习的网络模型；将需要去雨的图像输入到训练好的基于图像先验和门控注意力学习的网络模型,最终获取去雨后的图像数据,本发明有效的提取图像的细节信息和识别出更多的雨条纹,从而增强了雨图的去雨效果。
- 单位
