[發明專利]水印去除模型訓練方法、裝置、計算機設備和存儲介質

在審

• 申請號: 202011238056.2
• 申請日: 2020-11-09
• 公開（公告）號: CN112330522A
• 公開（公告）日: 2021-02-05
• 發明（設計）人: 張少林;寧欣;曾慶亮;許少輝
• 申請（專利權）人: 深圳市威富視界有限公司
• 主分類號: G06T1/00
• 分類號: G06T1/00;G06T9/00;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 廣州華進聯合專利商標代理有限公司 44224
• 代理人: 吳平

本申請涉及一種水印去除模型訓練方法、裝置、計算機設備和存儲介質。方法包括：在樣本圖像數據集中提取水印圖像以及對應的干凈圖像；將提取的圖像輸入至待訓練水印去除模型中進行三次風格遷移，得到風格遷移結果，根據提取的圖像對風格遷移結果中的目標圖像進行識別，得到圖像識別結果；根據風格遷移結果、圖像識別結果、水印圖像以及干凈圖像計算待訓練水印去除模型對應的對抗損失值、循環一致性損失值以及身份重建損失值；根據上述各損失值計算待訓練水印去除模型對應的目標損失值；根據目標損失值對待訓練水印去除模型進行對抗訓練，直至達到預設條件，停止模型訓練，得到訓練后的水印去除模型。采用本方法能夠提高水印去除后的圖像質量。

技術領域

本申請涉及計算機技術領域，特別是涉及一種水印去除模型訓練方法、裝置、計算機設備和存儲介質。

背景技術

隨著數字媒體技術和計算機技術的發展，各種數字媒體如圖像，通過互聯網進行傳播，人們可以進行下載使用。為了保護圖像的版權，往往在圖像中添加水印。由于水印會在一定程度上干擾或破壞圖像的內在數據信息，為了更好地應用圖像的價值，需要將圖像中的水印去除。

目前，可以通過生成式對抗模型對水印圖像進行水印去除，得到對應的干凈圖像，然而，傳統的生成式對抗模型在水印去除過程中，可能會丟失水印圖像的原有信息，導致得到的干凈圖像的質量較低。

發明內容

基于此，有必要針對上述技術問題，提供一種能夠提高水印去除后的圖像質量的水印去除模型訓練方法、裝置、計算機設備和存儲介質。

一種水印去除模型訓練方法，所述方法包括：

獲取樣本圖像數據集；

在所述樣本圖像數據集中提取水印圖像以及所述水印圖像對應的干凈圖像；

將所述水印圖像以及所述干凈圖像輸入至待訓練水印去除模型中，通過所述待訓練水印去除模型對所述水印圖像以及所述干凈圖像進行三次風格遷移，得到風格遷移結果，根據所述水印圖像以及所述干凈圖像對所述風格遷移結果中的目標圖像進行識別，得到圖像識別結果；

根據所述風格遷移結果、所述圖像識別結果、所述水印圖像以及所述干凈圖像計算所述待訓練水印去除模型對應的對抗損失值、循環一致性損失值以及身份重建損失值；

根據所述對抗損失值、所述循環一致性損失值以及所述身份重建損失值計算所述待訓練水印去除模型對應的目標損失值；

根據所述目標損失值對所述待訓練水印去除模型進行對抗訓練，直至達到預設條件，停止模型訓練，得到訓練后的水印去除模型。

在其中一個實施例中，所述將所述水印圖像以及所述干凈圖像輸入至待訓練水印去除模型中，通過所述待訓練水印去除模型對所述水印圖像以及所述干凈圖像進行三次風格遷移，得到風格遷移結果包括：

將所述水印圖像以及所述干凈圖像輸入至待訓練水印去除模型的對應生成器中進行三次風格遷移，得到所述水印圖像對應的無水印圖像、所述干凈圖像對應的有水印圖像、所述無水印圖像對應的第一水印添加圖像以及所述有水印圖像對應的第一水印去除圖像、所述水印圖像對應的第二水印添加圖像和所述干凈圖像對應的第二水印去除圖像；

根據所述無水印圖像、所述有水印圖像、所述第一水印添加圖像、所述第一水印去除圖像、所述第二水印添加圖像和所述第二水印去除圖像生成風格遷移結果。

在其中一個實施例中，所述待訓練水印去除模型的生成器包括第一生成器和第二生成器，所述將所述水印圖像以及所述干凈圖像輸入至待訓練水印去除模型的對應生成器中進行三次風格遷移，得到所述水印圖像對應的無水印圖像、所述干凈圖像對應的有水印圖像、所述無水印圖像對應的第一水印添加圖像以及所述有水印圖像對應的第一水印去除圖像、所述水印圖像對應的第二水印添加圖像和所述干凈圖像對應的第二水印去除圖像包括：