[发明专利]一种对抗网络的训练方法、装置、电子设备和存储介质

权利要求书

1.一种对抗网络的训练方法，其特征在于，所述对抗网络包括待训练的生成器和待训练判别器，所述方法包括：

获取训练集，所述训练集包括N张融合图片，所述N张融合图片中的每张融合图片是根据N张第一图片中的每张第一图片和N张第二图片中的每张第二图片确定的，所述N张第一图片中的每张第一图片为只包含水印的图片，所述N张第二图片中的每张第二图片为不包含水印的图片，其中，所述N为大于0的整数；

将所述N张融合图片输入所述待训练的生成器，以得到N张生成图片，所述N张生成图片中的每张生成图片均不包含水印；

采用损失函数交替更新所述待训练的生成器和所述待训练判别器，所述损失函数是根据所述N张第一图片、所述N张第二图片和所述N张生成图片确定的。

2.根据权利要求1所述的方法，其特征在于，融合图片A为所述N张融合图片中的任意一张图片，所述获取训练集，包括：

获取所述融合图片A对应的第一图片和所述融合图片A对应的第二图片，所述融合图片A对应的第一图片为所述N张第一图片中的一张图片，所述融合图片A对应的第二图片为所述N张第二图片中的一张图片；

采用第一公式对所述融合图片A对应的第一图片和所述融合图片A对应的第二图片进行处理，得到所述融合图片A；

将所述融合图片A确定为所述训练集中的一张图片；

其中，所述第一公式是根据所述融合图片A对应的第一图片中像素的灰度值和所述融合图片A对应的第二图片中像素的灰度值确定的。

3.根据权利要求2所述的方法，其特征在于，所述采用第一公式对所述融合图片A对应的第一图片和所述融合图片A对应的第二图片进行处理，得到所述融合图片A，包括：

根据所述融合图片A对应的第一图片，确定第一像素矩阵，所述第一像素矩阵包括所述融合图片A对应的第一图片中每个像素对应的灰度值；

检测所述第一像素矩阵中是否存在灰度值为预设灰度值的像素；

若是，则确定所述第一像素矩阵中每列像素的灰度值为所述预设灰度值的像素的个数，按照第一顺序，采用所述第一公式对所述融合图片A对应的第一图片和所述融合图片A对应的第二图片进行处理，得到所述融合图片A，所述第一顺序是根据所述第一像素矩阵中每列像素的灰度值为所述预设灰度值的像素的个数确定的；

若否，则根据所述融合图片A对应的第二图片，确定第二像素矩阵，所述第二像素矩阵包括所述融合图片A对应的第二图片中每个像素对应的灰度值；若所述第二像素矩阵中存在灰度值为所述预设灰度值的像素，则确定所述第二像素矩阵中每列像素的灰度值为所述预设灰度值的像素的个数，按照第二顺序，采用所述第一公式对融合图片A对应的第一图片和融合图片A对应的第二图片进行处理，得到融合图片A，所述第二顺序是根据所述第二像素矩阵中每列像素的灰度值为所述预设灰度值的像素的个数确定的。

4.根据权利要3所述的方法，其特征在于，所述按照第一顺序，采用所述第一公式对所述融合图片A对应的第一图片和所述融合图片A对应的第二图片进行处理，得到所述融合图片A，包括：

根据所述第一像素矩阵中每列像素的灰度值为所述预设灰度值的像素的个数，确定所述第一像素矩阵中的至少一列像素P1，第一列像素为所述至少一列像素P1中的任意一列，所述第一列像素中每个像素的灰度值为所述预设灰度值的像素的个数小于阈值；

按照所述至少一列像素P1中的每列像素的灰度值为所述预设灰度值的像素的个数从小到大的顺序，采用所述第一公式对所述至少一列像素P1和至少一列像素Q1进行处理，得到至少一列融合像素K1，所述至少一列像素Q1为所述第二像素矩阵中与所述至少一列像素P1对应的像素；

在检测到所述至少一列像素P1中最后一列像素处理完成后，采用所述第一公式并行对所述第一像素矩阵中除所述至少一列像素P1之外的其他像素和所述第二像素矩阵中除所述至少一列像素Q1之外的其他像素进行处理，得到至少一列融合像素K2；

根据所述至少一列融合像素K1和所述至少一列融合像素K2，确定所述融合图片A。