基于雨条纹指导的单幅图像的渐进式去雨方法|科研之友

摘要

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

摘要

关键词