一种雨密度分类引导的双阶段单幅图像去雨方法

摘要

本发明提出了一种雨密度分类引导的双阶段单幅图像去雨方法；本发明引入多幅合成雨天图像和对应的真实清晰图像，进一步引入每幅合成雨天图像对应的雨线层图像、全局大气光图像、传输地图图像、雨密度分类标签真值；依次引入先验去雨网络、雨密度分类网络和优化去雨网络，结合先前所引入的数据分别构建损失函数，得到训练好的各网络模型；将真实雨天图像输入训练好的先验去雨模型和雨密度分类模型，得到对应的预测初始去雨背景和预测雨密度分类级别，再将它们输入至训练好的优化去雨模型得到对应的去雨图像；本发明能解决现有技术难以同时去除雨线和雨雾、对不同密度、不同模式的雨处理不够鲁棒的问题，能对真实雨天场景进行高效、清晰地复原。