[發明專利]一種水印分類模型訓練方法及裝置

有效

• 申請號: 202110313604.1
• 申請日: 2021-03-24
• 公開（公告）號: CN113111734B
• 公開（公告）日: 2022-05-17
• 發明（設計）人: 宮明明;鄭霖;曾小英;陳虹
• 申請（專利權）人: 支付寶（杭州）信息技術有限公司
• 主分類號: G06V10/764
• 分類號: G06V10/764;G06V10/774;G06V10/82;G06V30/413;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京博思佳知識產權代理有限公司 11415
• 代理人: 周嗣勇

本說明書提出一種水印分類模型訓練方法及裝置，預先訓練水印檢測模型以及文本檢測模型，所述水印檢測模型用于從圖像中檢測出預設類型的水印、以及每個預設類型水印的位置；所述文本檢測模型用于從圖像中檢測出文本信息，以及每條文本信息的位置；所述方法包括：獲取訓練圖像數據集，所述數據集中任一訓練圖像中包括至少一個水印；所述水印的類型包括合法水印或非法水印；利用所述訓練圖像數據集、所述水印檢測模型、和所述文本檢測模型，構建水印分類模型的訓練樣本集；利用所述訓練樣本集對初始水印分類模型進行有監督訓練，得到用于進行水印分類的水印分類模型。

技術領域

本說明書涉及機器學習技術領域，尤其涉及一種水印分類模型訓練方法及裝置。

背景技術

互聯網平臺往往需要對用戶上傳的圖片進行審核，例如會在商戶注冊時，要求商戶上傳一些店鋪內的實景圖像，以用于審核身份。但在實際應用中發現存在商戶盜用網絡圖像作為實景圖像、以欺騙審核的情況，因此需要對實景圖像以及網絡圖像加以區分，而網絡圖像的一個顯著特征是包含水印，因此可以根據圖像是否包含水印進行區分。

另外，由于部分拍攝應用在拍攝圖像后，也會在圖像中增加水印，因此需要區分合法水印和非法水印。合法水印可以包括拍攝應用的水印、時間戳水印等，具有這類水印的圖像一般可以認為是用戶自行提供的。而非法水印則包括各種網站的水印、公眾號水印、用戶名水印等，具有這類水印的圖像往往是用戶從別處下載的，因此，需要對圖像中不同的水印進行分類，從而利用分類結果對用戶上傳的圖片進行審核。

目前，常采用人工分類的方式對水印進行分類，但這種方式效率較低，因此，在對用戶上傳的圖像中的水印進行分類時，如何能夠提供效率且能保證分類準確性是目前需要解決的問題。

發明內容

針對上述技術問題，本說明書提供一種水印分類模型訓練方法及裝置，技術方案如下：

根據本說明書的第一方面，提供一種水印分類模型訓練方法，預先訓練水印檢測模型以及文本檢測模型，所述水印檢測模型用于從圖像中檢測出預設類型的水印、以及每個預設類型水印的位置；所述文本檢測模型用于從圖像中檢測出文本信息，以及每條文本信息的位置；所述方法包括：

獲取訓練圖像數據集，所述數據集中任一訓練圖像中包括至少一個水印；所述水印的類型包括合法水印或非法水印；

利用所述訓練圖像數據集、所述水印檢測模型、和所述文本檢測模型，構建水印分類模型的訓練樣本集；其中，任一訓練樣本，以目標水印圖像和該目標水印圖像中的文本信息為特征值，以該目標水印的類型為標簽值；

利用所述訓練樣本集對初始水印分類模型進行有監督訓練，得到用于進行水印分類的水印分類模型。

根據本說明書的第二方面，提供一種基于上述水印分類模型的水印分類方法，預先訓練水印檢測模型以及文本檢測模型，所述水印檢測模型用于從圖像中檢測出預設類型的水印、以及每個預設類型水印的位置；所述文本檢測模型用于從圖像中檢測出文本信息，以及每條文本信息的位置；所述方法包括：

獲取待審核圖像；

利用所述水印檢測模型和所述文本檢測模型，從所述待審核圖像中獲取至少一個待分類水印圖像，以及所述至少一個待分類水印圖像分別包含的文本信息；

針對任一待分類水印圖像，將該待分類水印圖像，以及其中的文本信息輸入到所述水印分類模型中，利用所述水印分類模型的輸出結果確定該待分類水印圖像中水印的類型。

根據本說明書的第三方面，提供一種水印分類模型的訓練裝置，預先訓練水印檢測模型以及文本檢測模型，所述水印檢測模型用于從圖像中檢測出預設類型的水印、以及每個預設類型水印的位置；所述文本檢測模型用于從圖像中檢測出文本信息，以及每條文本信息的位置；所述裝置包括：