[发明专利]基于水印数据增强的图片隐形水印检测方法

权利要求书

1.一种基于水印数据增强的图片隐形水印检测方法，其特征在于：包括如下步骤：

获得训练好的水印增强网络和分类网络；

将待检测图片输入训练好的水印增强网络中得到水印概率图；

将待检测图片和水印概率图进行通道融合，并将融合后的四通道图输入至训练好的分类网络中得到分类结果，分类结果为有水印或无水印。

2.如权利要求1所述的基于水印数据增强的图片隐形水印检测方法，其特征在于：所述的获得训练好的水印增强网络和分类网络包括：

构建水印增强网络和分类网络；

分别准备第一样本数据集和第二样本数据集；

分别确定两个网络的损失函数；

利用第一样本数据集训练水印增强网络得到训练好的水印增强网络，利用第二样本数据集训练分类网络得到训练好的分类网络。

3.如权利要求2所述的基于水印数据增强的图片隐形水印检测方法，其特征在于：所述的分别准备第一样本数据集和第二样本数据集包括如下步骤：

准备无水印的图片作为负样本图片，负样本图片对应的水印概率图中任一像素均标记为0；

根据水印生成算法，对无水印的图片进行水印嵌入操作，得到带隐形水印的正样本图片，并根据生成算法直接生成正样本图片的水印概率图，正样本图片对应的水印概率图中含水印的像素区域标记为1、无水印的像素区域标记为0；

将正样本图片、负样本图片分别与其对应的水印概率图进行通道融合得到四通道图；

正样本图片及其对应的水印概率图构成第一样本数据集；四通道图及其对应的类别标签构成第二样本数据集。

4.如权利要求3所述的基于水印数据增强的图片隐形水印检测方法，其特征在于：对所述的正样本图片和负样本图片按如下任一种方法进行处理：图像剪裁拼接处理、边缘镜像填充处理、压缩处理、高斯噪音以及高斯模糊滤波处理。

5.如权利要求1所述的基于水印数据增强的图片隐形水印检测方法，其特征在于：所述的水印增强网络以Unet网络为基础，并按如下步骤进行改进：

将Unet网络中的max pool操作替换为空洞抽样池化加第一1*1的卷积层，空洞抽样池化将C1×H1×W1大小的特征图处理成(4×C1)×(H1/2)×(W1/2)大小，第一1*1卷积层将(4×C1)×(H1/2)×(W1/2)大小的特征图降低为C1×(H1/2)×(W1/2)大小；

将Unet网络中的上采样操作替换为第二1*1卷积层加空洞上采样操作，第二1*1卷积层用于扩充特征图的通道数至原通道数的4倍，空洞上采样操作为空洞抽样池化操作的逆处理。

6.一种计算机可读存储介质，其特征在于：其上存储有计算机程序，所述计算机程序被处理器执行时，实现如权利要求1-5中任一项所述的基于水印数据增强的图片隐形水印检测方法。

7.一种电子设备，其特征在于：包括存储器、处理器及存储在存储器上的计算机程序，所述处理器执行所述计算机程序时，实现如权利要求1-5中任一项所述的基于水印数据增强的图片隐形水印检测方法。