基于视觉质量评价反馈的图像去雨方法及电子装置

摘要

本发明公开了一种基于视觉质量评价反馈的图像去雨方法及电子装置，包括基于若干样本无雨图像及生成的雨痕与雨雾，构建成对图像数据集；将搜集的样本雨天图像进行去雨处理，并对各处理后样本雨天图像进行人工视觉质量评估，构建非成对雨天图像质量数据集；利用非成对雨天图像质量数据集，训练第一卷积神经网络，得到质量评估网络；使用成对图像数据集和质量评估网络约束，训练第二卷积神经网络，得到图像去雨模型；将待处理图像输入图像去雨模型，得到去雨后的图像。本发明使用了真实的雨天图像参与模型训练，使模型能学习到处理更加丰富和真实的降质类型，引入了视觉质量评价反馈，使生成的去雨图像具有人眼主观意义上更好的质量。