科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于雨条纹指导的单幅图像的渐进式去雨方法,包括以下步骤：S1：构建具有多个阶段的渐进式去雨网络,每一阶段均包括渐进式雨条纹检测模块和渐进式雨条纹去除模块；S2：获取有雨图像,将有雨图像输入渐进式去雨网络进行训练,渐进式雨条纹检测模块输出雨条纹掩膜,渐进式雨条纹去除模块输出去雨图像；S3：设定渐进式去雨网络的组合损失函数,根据组合损失函数对渐进式去雨网络进行优化,获得渐进式去雨网络模型；S4：将待去雨的有雨图像输入训练好的渐进式去雨网络模型,渐进式去雨网络模型输出去除雨条纹后的无雨图像。本发明可以有效地恢复被雨条纹遮挡的图像,解决雨条纹去除不干净的情况,实现了每阶段逐步去除雨条纹的效果。
- 单位
