[发明专利]一种自动去除图片水印的方法及装置

权利要求书

1.一种自动去除图片水印的方法，其特征在于，包括：

获取带水印图片；

将所述带水印图片输入训练好的预设神经网络，根据所述训练好的预设神经网络的输出结果，得到去水印图片。

2.根据权利要求1所述自动去除图片水印的方法，其特征在于，在所述获取带水印图片的步骤之后，所述方法还包括：

对带水印图片进行识别，确定水印类别信息。

3.根据权利要求2所述自动去除图片水印的方法，其特征在于，在将所述带水印图片输入训练好的预设神经网络的步骤之前，所述方法还包括：

获取无水印样本图片集信息和水印类别信息；

根据所述水印类别信息对无水印样本集图片信息进行水印添加处理，得到带水印样本图片集信息。

4.根据权利要求3所述自动去除图片水印的方法，其特征在于，在所述得到带水印样本图片信息的步骤之后，所述方法还包括：

获取无水印样本图片集信息和带水印样本图片集信息；

神经网络生成器对所述带水印样本图片集信息进行去水印处理，得到生成样本图片集信息；

神经网络分类器根据所述无水印样本图片集信息对所述生成样本图片集信息进行分类识别，得到分类结果；

根据所述神经网络生成器和神经网络分类器进行对抗学习，当满足预设条件时，得到训练好的预设神经网络。

5.根据权利要求3所述自动去除图片水印的方法，其特征在于，所述根据所述水印类别信息对无水印样本集图片信息进行水印添加处理，得到带水印样本图片集信息的步骤，具体包括：

根据Canny算法和水印类别信息，获取水印形状信息；

根据所述水印形状信息和水印添加算法得到水印特征信息；

根据所述水印特征信息对所述无水印样本集图片信息进行水印添加处理，得到带水印样本图片集信息。

6.根据权利要求5所述自动去除图片水印的方法，其特征在于，所述水印添加算法具体为：

其中，J(x,y)是生成的带有水印图片的(x,y)坐标颜色特征值，W是现有且待添加水印的图像矩阵，I是待添加水印特征矩阵，α则为水印与图片叠加的权重系数。

7.根据权利要求1所述自动去除图片水印的方法，其特征在于，所述方法还包括：

将带水印图片划分为多个带水印原始子图片；

将所述多个带水印原始子图片依次输入训练好的预设神经网络，根据所述训练好的预设神经网络的输出结果，得到多个去水印子图片；

将所述多个去水印子图片进行合成处理，得到去水印图片。

8.一种自动去除图片水印的装置，其特征在于，包括：

获取模块，用于获取带水印图片；

去除模块，用于将所述带水印图片输入训练好的预设神经网络，根据所述训练好的预设神经网络的输出结果，得到去水印图片。

9.一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至7任一项所述自动去除图片水印的方法的步骤。

10.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时实现如权利要求1至7任一项所述自动去除图片水印方法的步骤。