[发明专利]基于生成对抗网络的图像分块的单张图像去雨方法

摘要

本发明提供了一种基于生成对抗网络的图像分块的单张图像去雨方法，通过将图像划分成互不重叠的大小相同的图像块，将每个图像块作为条件生成对抗网络的输入，降低输入的维度；一个生成对抗网络被训练用来实现有雨图像块到无雨图像块的非线性映射，有效克服了许多细节部分被忽略的问题，并且可以尽可能地在每个尺度上去除雨线；为了更好地维持去雨图像块之间的结构、颜色等方面的一致性，引用了双边滤波器和非均值局部去噪算法，构建了新的误差函数，添加至条件生成对抗网络的总的误差函数中。本发明不需要任何先验知识，也不需要对图像进行预处理和后处理，保证了整个结构的完整性。在测试集上的结果显示本发明先比经典算法提高了4～7dB。

主权项

1.基于生成对抗网络的图像分块的单张图像去雨方法，其特征在于，包括以下步骤：第S1步，建立去雨图像数据库；第S2步，建立与去雨图像数据库相对应的去雨图像块数据库；第S3步，将去雨图像块数据库作为条件生成对抗网络的输入，利用去雨图像数据库训练条件生成对抗网络，实现去雨图像块数据库中有雨图像块至无雨图像块的非线性映射；同时将条件生成对抗网络的原对抗误差函数修改为添加了均方误差、去噪预测误差和视觉误差的权重误差函数；第S4步，将条件生成对抗网络生成的所有尺度上的无雨图像块整合在一起，形成去雨后无雨图像。