[发明专利]一种从去除到渲染的双向控雨图像处理方法

摘要

本发明公开一种从去除到渲染的双向控雨图像处理方法。其方法的步骤为：构造包括训练集和测试集的数据库；将雨图输入背景提取网络和高频雨去除网络，分别输出包含背景信息的背景图和包含背景及雾信息的雾图；构建主控制网络的损失函数，该损失函数包括方向梯度直方图损失函数和自相关损失函数；输入雨图、背景图、雾图到主控制网络，用多分支结构训练输出从无雨到大雨的不同等级雨图；在测试阶段通过特征空间插值，可得到任意中间不同雨量强度等级的雨图。本发明的方法可以实现输入单张雨图，由用户自己控制得到从完全无雨到大雨之间的任意中间强度雨图，且在此变化过程中，雨纹的方向、外观、分布等场景特定的雨特征被保留。