[发明专利]一种基于场景深度的单图像去雨雾方法、设备及存储介质

摘要

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