科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于场景深度的单图像去雨雾方法、设备及存储介质,包括：构建去雨雾任务训练网络,所述去雨雾任务训练网络包括第一生成器、第二生成器及判别器,所述第一生成器输入为雨图,输出为单通道的场景深度图,所述场景深度图与雨图进行拼接后输入第二生成器,输出去雨雾图片；所述判别器用于判断生成的去雨雾图片与标签雨图是否一致；获取训练样本,所述训练样本包括雨图,雨图对应场景深度图及干净无雨图；对去雨雾任务训练网络进行训练,得到去雨雾生成模型。本发明对雨水图片处理操作简单,只需要单张雨水图片,经过训练好的生成式网络前向传播就能得到去雨后的干净图片。
- 单位
