[發(fā)明專利]一種對抗網(wǎng)絡(luò)的訓(xùn)練方法、裝置、電子設(shè)備和存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202011070284.3
• 申請日: 2020-09-30
• 公開（公告）號: CN112215276A
• 公開（公告）日: 2021-01-12
• 發(fā)明（設(shè)計(jì)）人: 李國安
• 申請（專利權(quán)）人: 深圳壹賬通智能科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06T5/50;G06N3/04;G06N3/08
• 代理公司: 廣州三環(huán)專利商標(biāo)代理有限公司 44202
• 代理人: 熊永強(qiáng)

本申請涉及模型構(gòu)建，公開了一種對抗網(wǎng)絡(luò)的訓(xùn)練方法、裝置、電子設(shè)備和存儲介質(zhì)，該方法包括：獲取訓(xùn)練集，訓(xùn)練集包括N張融合圖片，N張融合圖片中的每張融合圖片是根據(jù)N張第一圖片中的每張第一圖片和N張第二圖片中的每張第二圖片確定的，N張第一圖片中的每張第一圖片為只包含水印的圖片，N張第二圖片中的每張第二圖片為不包含水印的圖片；將N張融合圖片輸入待訓(xùn)練的生成器，以得到N張生成圖片，N張生成圖片中的每張生成圖片均不包含水印；采用損失函數(shù)交替更新待訓(xùn)練的生成器和待訓(xùn)練判別器，損失函數(shù)是根據(jù)N張第一圖片、N張第二圖片和N張生成圖片確定的。實(shí)施本申請實(shí)施例，實(shí)現(xiàn)了精準(zhǔn)去除水印。

技術(shù)領(lǐng)域

本申請涉及計(jì)算機(jī)技術(shù)領(lǐng)域，尤其涉及一種對抗網(wǎng)絡(luò)的訓(xùn)練方法、裝置、電子設(shè)備和存儲介質(zhì)。

背景技術(shù)

目前，為了減少對有水印的圖片的識別錯誤率，往往會涉及到采用模型來去除水印。比如，為了減少對有水印的廣告圖片、有水印的賬單截圖等，一般采用模型來去除水印，以讓用戶更好的使用或者查看廣告圖片、賬單截圖等。因此，在利用模型去除水印時，會涉及到模型的訓(xùn)練。在進(jìn)行模型訓(xùn)練時，會采用人工標(biāo)注好的訓(xùn)練集。

一般來說，人工標(biāo)注好的訓(xùn)練集中的每張圖片均是已標(biāo)注好水印位置的圖片。在標(biāo)注時可能出現(xiàn)標(biāo)注失誤的情況，當(dāng)利用這樣的訓(xùn)練集訓(xùn)練模型后，可能導(dǎo)致訓(xùn)練好的模型出現(xiàn)無法精準(zhǔn)去除水印的問題。

發(fā)明內(nèi)容

本申請實(shí)施例提供了一種對抗網(wǎng)絡(luò)的訓(xùn)練方法、裝置、電子設(shè)備和存儲介質(zhì)，實(shí)施本申請實(shí)施例，實(shí)現(xiàn)了精準(zhǔn)去除水印。

本申請第一方面提供了一種對抗網(wǎng)絡(luò)的訓(xùn)練方法，所述對抗網(wǎng)絡(luò)包括待訓(xùn)練的生成器和待訓(xùn)練判別器，所述方法包括：

獲取訓(xùn)練集，所述訓(xùn)練集包括N張融合圖片，所述N張融合圖片中的每張融合圖片是根據(jù)N張第一圖片中的每張第一圖片和N張第二圖片中的每張第二圖片確定的，所述N張第一圖片中的每張第一圖片為只包含水印的圖片，所述N張第二圖片中的每張第二圖片為不包含水印的圖片，其中，所述N為大于0的整數(shù)；

將所述N張融合圖片輸入所述待訓(xùn)練的生成器，以得到N張生成圖片，所述N張生成圖片中的每張生成圖片均不包含水印；

采用損失函數(shù)交替更新所述待訓(xùn)練的生成器和所述待訓(xùn)練判別器，所述損失函數(shù)是根據(jù)所述N張第一圖片、所述N張第二圖片和所述N張生成圖片確定的。

本申請第二方面提供了一種對抗網(wǎng)絡(luò)的訓(xùn)練裝置，所述對抗網(wǎng)絡(luò)包括待訓(xùn)練的生成器和待訓(xùn)練判別器，所述裝置包括獲取模塊、輸入模塊和更新模塊，其中，

所述獲取模塊，用于獲取訓(xùn)練集，所述訓(xùn)練集包括N張融合圖片，所述N張融合圖片中的每張融合圖片是根據(jù)N張第一圖片中的每張第一圖片和N張第二圖片中的每張第二圖片確定的，所述N張第一圖片中的每張第一圖片為只包含水印的圖片，所述N張第二圖片中的每張第二圖片為不包含水印的圖片，其中，所述N為大于0的整數(shù)；

所述輸入模塊，用于將所述N張融合圖片輸入所述待訓(xùn)練的生成器，以得到N張生成圖片，所述N張生成圖片中的每張生成圖片均不包含水印；

所述更新模塊，用于采用損失函數(shù)交替更新所述待訓(xùn)練的生成器和所述待訓(xùn)練判別器，所述損失函數(shù)是根據(jù)所述N張第一圖片、所述N張第二圖片和所述N張生成圖片確定的。

本申請第三方面提供了一種對抗網(wǎng)絡(luò)的訓(xùn)練的電子設(shè)備，包括處理器、存儲器、通信接口以及一個或多個程序，其中，所述一個或多個程序被存儲在所述存儲器中，并且被生成由所述處理器執(zhí)行，以執(zhí)行一種對抗網(wǎng)絡(luò)的訓(xùn)練方法任一項(xiàng)方法中的步驟的指令。

本申請第四方面提供了一種計(jì)算機(jī)可讀存儲介質(zhì)，所述計(jì)算機(jī)可讀存儲介質(zhì)用于存儲計(jì)算機(jī)程序，所述存儲計(jì)算機(jī)程序被所述處理器執(zhí)行，以實(shí)現(xiàn)一種對抗網(wǎng)絡(luò)的訓(xùn)練方法任一項(xiàng)所述的方法。