[发明专利]图像去雨方法、系统、计算机设备及介质

权利要求书

1.一种图像去雨方法，其特征在于，包括：

选取多张覆盖雨信号的图像及其对应的真实去雨图像作为训练集；

在所述训练集上对总损失函数进行训练，获得具有对偶判别器的生成式对抗网络；

采用所述生成式对抗网络中的生成器对真实雨图进行去雨，生成对应的去雨图；

在所述训练集上对总损失函数进行训练的步骤之前，所述图像去雨方法还包括：

根据均方误差损失函数、感知损失函数及对偶对抗损失函数构建所述总损失函数；

所述均方误差损失函数为：

所述感知损失函数为：

所述对偶对抗损失函数为：

其中size(·)表示所述覆盖雨信号的图像的大小，G(·)表示所述生成式对抗网络中的生成器，V(·)表示从所述生成式对抗网络输入到所述生成式对抗网络中全部卷积层的任一层输出的映射，D₁(·)为所述对偶判别器中用于鉴别生成的所述去雨图是否真实的判别器，D₂(·)为所述对偶判别器中用于鉴别生成的雨信号是否真实的判别器，I_k为单位矩阵。

2.根据权利要求1所述的图像去雨方法，其特征在于，

所述总损失函数为：

L＝l_e+α_pl_p+α_al_a；

其中l_e为所述均方误差损失函数，l_p、α_p分别为所述感知损失函数及其对应的权重，l_a、α_a分别为所述对偶对抗损失函数及其对应的权重。

3.根据权利要求1所述的图像去雨方法，其特征在于，

所述训练集为：

其中X_k为所述覆盖雨信号的图像，Y_k为所述覆盖雨信号的图像对应的真实去雨图像。

4.一种图像去雨系统，其特征在于，包括：

选取单元，用于选取多张覆盖雨信号的图像及其对应的真实去雨图像作为训练集；

训练单元，用于在所述训练集上对总损失函数进行训练，获得具有对偶判别器的生成式对抗网络；

去雨单元，用于采用所述生成式对抗网络中的生成器对真实雨图进行去雨，生成对应的去雨图；

在所述训练集上对总损失函数进行训练的步骤之前，所述图像去雨系统还包括：

构建单元，用于根据均方误差损失函数、感知损失函数及对偶对抗损失函数构建所述总损失函数；

所述均方误差损失函数为：

所述感知损失函数为：

所述对偶对抗损失函数为：

其中size(·)表示所述覆盖雨信号的图像的大小，G(·)表示所述生成式对抗网络中的生成器，V(·)表示从所述生成式对抗网络输入到所述生成式对抗网络中全部卷积层的任一层输出的映射，D₁(·)为所述对偶判别器中用于鉴别生成的所述去雨图是否真实的判别器，D₂(·)为所述对偶判别器中用于鉴别生成的雨信号是否真实的判别器，I_k为单位矩阵。

5.根据权利要求4所述的图像去雨系统，其特征在于，

所述总损失函数为：

L＝l_e+α_pl_p+α_al_a；

其中l_e为所述均方误差损失函数，l_p、α_p分别为所述感知损失函数及其对应的权重，l_a、α_a分别为所述对偶对抗损失函数及其对应的权重。

6.根据权利要求4所述的图像去雨系统，其特征在于，

所述训练集为：

其中X_k为所述覆盖雨信号的图像，Y_k为所述覆盖雨信号的图像对应的真实去雨图像。