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

说明书

本发明实施例提供一种自动去除图片水印的方法及装置，该方法包括：获取带水印图片；将所述带水印图片输入训练好的预设神经网络，根据所述训练好的预设神经网络的输出结果，得到去水印图片。通过神经网络分类器对带水印图片集进行去除处理，生成去除水印后的样本图片集，将去除水印后的样本图片集、无水印图片集和带水印图片集作为训练样本，以去除水印后的样本图片集和无水印图片集的平均绝对误差值最小化为目标将神经网络生成器和神经网络分类器进行对抗学习，进行多次训练，得到能够有效去除水印的训练好的预设神经网络，从而可以根据训练好的预设神经网络对带水印图片进行去水印处理，高效的去水印图片。

技术领域

本发明涉及计算机技术领域，尤其涉及一种自动去除图片水印的方法及装置。

背景技术

随着网络技术的飞速发展，各种数字媒体，如图片通过互联网进行传播，为了表明图片版权所有人，在图片中叠加可见的水印记号是一种常见的做法，由于水印也是以图像的形式添加的，因此图像中的水印往往很难除去，但是很多时候图像的使用者需要使用没有水印的图像，此时则需要对图像进行去水印处理。

现有技术中对于图像进行去水印处理主要是通过人工通过图像编辑软件来进行，但是这种方法的效率比较低，很难实现对于大量水印图片的高效处理，因此如何实现对于大量水印图片的高效处理已经成为业界亟待解决的问题。

发明内容

本发明实施例提供一种自动去除图片水印的方法及装置，用以解决上述背景技术中提出的技术问题，或至少部分解决上述背景技术中提出的技术问题。

第一方面，本发明实施例提供一种自动去除图片水印的方法，包括：

获取带水印图片；

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

更具体的，在所述获取带水印图片的步骤之后，所述方法还包括：

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

更具体的，在将所述带水印图片输入训练好的预设神经网络的步骤之前，所述方法还包括：

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

对无水印样本集图片信息进行水印添加处理，得到带水印样本图片集信息。

更具体的，在所述得到带水印样本图片信息的步骤之后，所述方法还包括：

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

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

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

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

更具体的，所述对无水印样本集图片信息进行水印添加处理，得到带水印样本图片集信息的步骤，具体包括：

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

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

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

更具体的，所述水印添加算法具体为：

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