[发明专利]一种基于生成对抗网络的图像去雨滴方法

权利要求书

1.一种基于生成对抗网络的图像去雨滴方法，其特征在于，所述方法包括：

从数据库中获取外景图片集；

图像预处理，为所获取到的外景图片集加入下雨效果，构建训练集和测试集；

构建生成式网络，其输入为带雨场景图像，输出为清晰场景图像；

根据像素空间上的误差训练生成式网络；

加入特征空间上的误差再次训练生成式网络；

构建判别式网络，其输入为真实样本或者由生成器生成的样本，输出为真或假的单个标识；

把判别式网络加入到模型之中，采用误差反向传播算法训练所述的生成式网络；

将测试集中的带雨场景图输入训练好的生成式网络中，输出为对应的清晰场景图像。

2.如权利要求1所述的一种基于生成对抗网络的图像去雨滴方法，其特征在于，所述加入特征空间上的误差再次训练生成式网络，具体为：

引入已经训练好的比较器网络，把生成的清晰图片和实际的清晰图片分别输入到比较器网络中，分别获取两者在比较器网络中的特征图，计算两张特征图之间的欧氏距离，作为特征空间上的误差- ， 结合像素空间上的误差与特征空间上的误差作为总体误差，采用误差反向传播算法训练所述的生成式网络。