[发明专利]图像处理方法、训练方法、装置、设备及计算机可读存储介质

权利要求书

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

获取待处理图像，所述待处理图像为原始带雨图像；

第一生成器根据所述原始带雨图像生成纯雨图层，所述纯雨图层指示所述原始带雨图像的雨点位置；

第二生成器根据所述纯雨图层和所述原始带雨图像生成不带雨的背景图层。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：预先训练所述第一生成器和第二生成器，其中，所述预先训练所述第一生成器和第二生成器包括：

第一判别器获取第二生成器生成的所述不带雨的背景图层，得到对应的第一对抗性损失值GAN_Loss_B；

合成所述第二生成器生成的不带雨的背景图层和所述第一生成器生成的纯雨图层，得到合成的带雨图像；

第二判别器获取所述合成的带雨图像，得到对应的第二对抗性损失值GAN_Loss_A;

根据所述第一对抗性损失值GAN_Loss_B和所述第二对抗性损失值GAN_Loss_A确定第三对抗损失值，根据所述第三损失对抗值通过反向传播更新所述第一生成器和第二生成器的参数。

3.根据权利要求2所述的方法，其特征在于，第三对抗损失值根据所述第一对抗性损失值GAN_Loss_B和所述第二对抗性损失值GAN_Loss_A的加权平均获得。

4.根据权利要求2所述的方法，其特征在于，所述预先训练所述第一生成器和第二生成器还包括：

比较所述合成的带雨图像和所述原始带雨图像，获取循环一致损失值；

根据所述循环一致损失值通过反向传播更新所述第一生成器和第二生成器的参数。

5.一种图像处理装置，其特征在于，包括：

获取模块，用于获取待处理图像，所述待处理图像为原始带雨图像；

第一生成器，用于根据所述原始带雨图像生成纯雨图层，所述纯雨图层指示所述原始带雨图像的雨点位置；

第二生成器，用于根据所述纯雨图层和所述原始带雨图像生成不带雨的背景图层。

6.一种训练方法，其特征在于，包括：

获取原始带雨图像；

第一生成器根据所述原始带雨图像生成纯雨图层，所述纯雨图层指示所述原始带雨图像的雨点位置；

第二生成器根据所述纯雨图层和所述原始带雨图像生成不带雨的背景图层；

第一判别器获取第二生成器生成的所述不带雨的背景图层，得到对应的第一对抗性损失值GAN_Loss_B；

合成所述第二生成器生成的不带雨的背景图层和所述第一生成器生成的纯雨图层，得到合成的带雨图像；

第二判别器获取所述合成的带雨图像，得到对应的第二对抗性损失值GAN_Loss_A;

根据所述第一对抗性损失值GAN_Loss_B和所述第二对抗性损失值GAN_Loss_A确定第三对抗损失值，根据所述第三损失对抗值通过反向传播更新所述第一生成器和第二生成器的参数。

7.一种训练装置，其特征在于，还包括：

输入层，用于获取原始带雨图像；

第一生成器，用于根据所述原始带雨图像生成纯雨图层，所述纯雨图层指示所述原始带雨图像的雨点位置；

第二生成器，用于根据所述纯雨图层和所述原始带雨图像生成不带雨的背景图层；

第一判别器，用于获取所述第二生成器生成的所述不带雨的背景图层，得到对应的第一对抗性损失值GAN_Loss_B；

合成器，用于合成所述第二生成器生成的不带雨的背景图层和所述第一生成器生成的纯雨图层，得到合成的带雨图像；

第二判别器，用于获取所述合成的带雨图像，得到对应的第二对抗性损失值GAN_Loss_A;

第一参数调整模块，用于根据所述第一对抗性损失值GAN_Loss_B和所述第二对抗性损失值GAN_Loss_A确定第三对抗损失值，根据所述第三损失对抗值通过反向传播更新所述第一生成器和第二生成器的参数。