[发明专利]图像水印去除装置

权利要求书

1.一种图像水印去除装置，其特征在于，所述装置包括对抗生成网络模型，其配置为对人脸图像进行去水印处理；所述对抗生成网络模型包括生成器、判别器和优化模块；

所述生成器，配置为依据所述人脸图像生成对应的无水印图像；

所述判别器，配置为判断所述生成器所生成的无水印图像的整体图像和局部图像是否为真实图像；

所述优化模块，配置为按照下式所示的对抗损失函数优化所述生成器和判别器：

其中，所述loss_g和loss_d分别为所述生成器和判别器的损失函数，所述x₁为带水印人脸图像，所述[G(x₁)]⁺为对生成器G生成的无水印图像G(x₁)进行裁剪后得到的局部图像，所述y₁为所述带水印人脸图像x₁对应的无水印原始图像，所述y₂为所述无水印原始图像y₁中与所述[G(x₁)]⁺对应的局部图像，所述D₁(y₁)为判别器中全局判别单元D₁的输出结果，所述D₂(y₂)为判别器中局部判别单元D₂的输出结果，所述λ₁和λ₂均为预设的平衡系数。

2.根据权利要求1所述的装置，其特征在于，所述装置还包括人脸识别模型；

所述人脸识别模型，配置为提取所述无水印原始图像y₁和所述生成器生成的无水印图像G(x₁)中的人脸特征；

所述优化模块，进一步配置为按照下式所示的组合损失函数优化所述生成器和判别器：

其中，所述G_loss和D_loss分别为所述生成器和判别器的组合损失函数，所述loss_feature为依据所述人脸识别模型所提取的人脸特征构建的特征损失函数，所述loss_pixel为依据所述带水印人脸图像x₁与无水印原始图像y₁的像素差所构建的像素损失函数，所述γ₁、γ₂、γ₃、γ₄和γ₅均为预设的平衡系数。

3.根据权利要求2所述的装置，其特征在于，所述特征损失函数loss_feature如式所示：

loss_feature＝||φ(y₁)-φG(x₁)||₂

其中，所述φ(y₁)为人脸识别模型φ对无水印原始图像y₁提取得到的人脸特征，所述φG(x₁)为人脸识别模型φ对无水印图像G(x₁)提取得到的人脸特征，所述||||₂表示2范数。

4.根据权利要求2所述的装置，其特征在于，所述像素损失函数loss_pixel如下式所示：

loss_pixel＝||y₁-x₁||₂

其中，所述||||₂表示2范数。