[发明专利]一种水印分类模型训练方法及装置

说明书

本说明书提出一种水印分类模型训练方法及装置，预先训练水印检测模型以及文本检测模型，所述水印检测模型用于从图像中检测出预设类型的水印、以及每个预设类型水印的位置；所述文本检测模型用于从图像中检测出文本信息，以及每条文本信息的位置；所述方法包括：获取训练图像数据集，所述数据集中任一训练图像中包括至少一个水印；所述水印的类型包括合法水印或非法水印；利用所述训练图像数据集、所述水印检测模型、和所述文本检测模型，构建水印分类模型的训练样本集；利用所述训练样本集对初始水印分类模型进行有监督训练，得到用于进行水印分类的水印分类模型。

技术领域

本说明书涉及机器学习技术领域，尤其涉及一种水印分类模型训练方法及装置。

背景技术

互联网平台往往需要对用户上传的图片进行审核，例如会在商户注册时，要求商户上传一些店铺内的实景图像，以用于审核身份。但在实际应用中发现存在商户盗用网络图像作为实景图像、以欺骗审核的情况，因此需要对实景图像以及网络图像加以区分，而网络图像的一个显著特征是包含水印，因此可以根据图像是否包含水印进行区分。

另外，由于部分拍摄应用在拍摄图像后，也会在图像中增加水印，因此需要区分合法水印和非法水印。合法水印可以包括拍摄应用的水印、时间戳水印等，具有这类水印的图像一般可以认为是用户自行提供的。而非法水印则包括各种网站的水印、公众号水印、用户名水印等，具有这类水印的图像往往是用户从别处下载的，因此，需要对图像中不同的水印进行分类，从而利用分类结果对用户上传的图片进行审核。

目前，常采用人工分类的方式对水印进行分类，但这种方式效率较低，因此，在对用户上传的图像中的水印进行分类时，如何能够提供效率且能保证分类准确性是目前需要解决的问题。

发明内容

针对上述技术问题，本说明书提供一种水印分类模型训练方法及装置，技术方案如下：

根据本说明书的第一方面，提供一种水印分类模型训练方法，预先训练水印检测模型以及文本检测模型，所述水印检测模型用于从图像中检测出预设类型的水印、以及每个预设类型水印的位置；所述文本检测模型用于从图像中检测出文本信息，以及每条文本信息的位置；所述方法包括：

获取训练图像数据集，所述数据集中任一训练图像中包括至少一个水印；所述水印的类型包括合法水印或非法水印；

利用所述训练图像数据集、所述水印检测模型、和所述文本检测模型，构建水印分类模型的训练样本集；其中，任一训练样本，以目标水印图像和该目标水印图像中的文本信息为特征值，以该目标水印的类型为标签值；

利用所述训练样本集对初始水印分类模型进行有监督训练，得到用于进行水印分类的水印分类模型。

根据本说明书的第二方面，提供一种基于上述水印分类模型的水印分类方法，预先训练水印检测模型以及文本检测模型，所述水印检测模型用于从图像中检测出预设类型的水印、以及每个预设类型水印的位置；所述文本检测模型用于从图像中检测出文本信息，以及每条文本信息的位置；所述方法包括：

获取待审核图像；

利用所述水印检测模型和所述文本检测模型，从所述待审核图像中获取至少一个待分类水印图像，以及所述至少一个待分类水印图像分别包含的文本信息；

针对任一待分类水印图像，将该待分类水印图像，以及其中的文本信息输入到所述水印分类模型中，利用所述水印分类模型的输出结果确定该待分类水印图像中水印的类型。

根据本说明书的第三方面，提供一种水印分类模型的训练装置，预先训练水印检测模型以及文本检测模型，所述水印检测模型用于从图像中检测出预设类型的水印、以及每个预设类型水印的位置；所述文本检测模型用于从图像中检测出文本信息，以及每条文本信息的位置；所述装置包括：