[發(fā)明專(zhuān)利]一種網(wǎng)絡(luò)模型的訓(xùn)練方法以及相關(guān)裝置

有效

• 申請(qǐng)?zhí)枺?/td>: 201910331313.8
• 申請(qǐng)日: 2019-04-23
• 公開(kāi)（公告）號(hào): CN110222705B
• 公開(kāi)（公告）日: 2023-10-24
• 發(fā)明（設(shè)計(jì)）人: 陳漢亭;舒晗;王云鶴;許春景
• 申請(qǐng)（專(zhuān)利權(quán)）人: 華為技術(shù)有限公司
• 主分類(lèi)號(hào): G06V10/774
• 分類(lèi)號(hào): G06V10/774;G06V10/82;G06V10/74;G06N3/0475;G06N3/094;G06N3/0464;G06N3/045
• 代理公司: 深圳市深佳知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 44285
• 代理人: 王仲凱

1.一種網(wǎng)絡(luò)模型的訓(xùn)練方法，其特征在于，所述方法包括：

獲取第一生成網(wǎng)絡(luò)和第二生成網(wǎng)絡(luò)，其中，所述第一生成網(wǎng)絡(luò)為待訓(xùn)練的小型網(wǎng)絡(luò)，所述第二生成網(wǎng)絡(luò)為執(zhí)行過(guò)訓(xùn)練操作的大型網(wǎng)絡(luò)，所述第一生成網(wǎng)絡(luò)中的參數(shù)量小于所述第二生成網(wǎng)絡(luò)中的參數(shù)量；

將待轉(zhuǎn)換圖像輸入所述第一生成網(wǎng)絡(luò)，得到第一轉(zhuǎn)換后圖像；

將所述待轉(zhuǎn)換圖像輸入所述第二生成網(wǎng)絡(luò)，得到第二轉(zhuǎn)換后圖像；

根據(jù)所述第一轉(zhuǎn)換后圖像和所述第二轉(zhuǎn)換后圖像，對(duì)所述第一生成網(wǎng)絡(luò)進(jìn)行迭代訓(xùn)練，直至所述第一生成網(wǎng)絡(luò)輸出的所述第一轉(zhuǎn)換后圖像和所述第二生成網(wǎng)絡(luò)輸出的所述第二轉(zhuǎn)換后圖像的相似度達(dá)到所述預(yù)設(shè)閾值；

獲取第三生成網(wǎng)絡(luò)，所述第三生成網(wǎng)絡(luò)為所述第一生成網(wǎng)絡(luò)執(zhí)行過(guò)所述迭代訓(xùn)練后得到的網(wǎng)絡(luò)。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述根據(jù)所述第一轉(zhuǎn)換后圖像和所述第二轉(zhuǎn)換后圖像，對(duì)所述第一生成網(wǎng)絡(luò)進(jìn)行迭代訓(xùn)練，直至所述第一生成網(wǎng)絡(luò)輸出的所述第一轉(zhuǎn)換后圖像和所述第二生成網(wǎng)絡(luò)輸出的所述第二轉(zhuǎn)換后圖像的相似度達(dá)到預(yù)設(shè)閾值，包括：

獲取所述第一轉(zhuǎn)換后圖像的第一圖像矩陣；

獲取所述第二轉(zhuǎn)換后圖像的第二圖像矩陣；

根據(jù)所述第一圖像矩陣和所述第二圖像矩陣，采用第一損失函數(shù)對(duì)所述第一生成網(wǎng)絡(luò)進(jìn)行迭代訓(xùn)練，直至所述第一生成網(wǎng)絡(luò)輸出的所述第一轉(zhuǎn)換后圖像和所述第二生成網(wǎng)絡(luò)輸出的所述第二轉(zhuǎn)換后圖像的相似度達(dá)到所述預(yù)設(shè)閾值。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，

所述第一損失函數(shù)具體表現(xiàn)為：

其中，所述G_T(x)表示所述第一圖像矩陣，所述G_S(x)表示所述第二圖像矩陣，所述||G_T(x)-G_S(x)||₁表示所述第一圖像矩陣與所述第二圖像矩陣的差值的一范數(shù)，所述表示所述||G_T(x)-G_S(x)||₁的平方，所述L_L1(G_S)表示所述第一損失函數(shù)。

4.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述根據(jù)所述第一轉(zhuǎn)換后圖像和所述第二轉(zhuǎn)換后圖像，對(duì)所述第一生成網(wǎng)絡(luò)進(jìn)行迭代訓(xùn)練，直至所述第一生成網(wǎng)絡(luò)輸出的所述第一轉(zhuǎn)換后圖像和所述第二生成網(wǎng)絡(luò)輸出的所述第二轉(zhuǎn)換后圖像的相似度達(dá)到預(yù)設(shè)閾值，包括：

獲取第一判別網(wǎng)絡(luò)以及第二判別網(wǎng)絡(luò)，所述第一判別網(wǎng)絡(luò)為待訓(xùn)練網(wǎng)絡(luò)，所述第二判別網(wǎng)絡(luò)為執(zhí)行過(guò)訓(xùn)練操作的網(wǎng)絡(luò)，所述第一判別網(wǎng)絡(luò)與所述第一生成網(wǎng)絡(luò)歸屬于同一第一生成對(duì)抗網(wǎng)絡(luò)，所述第二判別網(wǎng)絡(luò)與所述第二生成網(wǎng)絡(luò)歸屬于同一第二生成對(duì)抗網(wǎng)絡(luò)；

將所述第一轉(zhuǎn)換后圖像輸入所述第一判別網(wǎng)絡(luò)，得到所述第一轉(zhuǎn)換后圖像的第一特征信息集合；

將所述第二轉(zhuǎn)換后圖像輸入所述第二判別網(wǎng)絡(luò)，得到所述第一轉(zhuǎn)換后圖像的第二特征信息集合；

根據(jù)所述第一特征信息集合和所述第二特征信息集合，采用第二損失函數(shù)對(duì)所述第一生成網(wǎng)絡(luò)進(jìn)行迭代訓(xùn)練，直至所述第一生成網(wǎng)絡(luò)輸出的所述第一轉(zhuǎn)換后圖像和所述第二生成網(wǎng)絡(luò)輸出的所述第二轉(zhuǎn)換后圖像的相似度達(dá)到所述預(yù)設(shè)閾值。

5.根據(jù)權(quán)利要求4所述的方法，其特征在于，

所述第二損失函數(shù)具體表現(xiàn)為：

其中，D_T(G_T(x))表示將所述第二轉(zhuǎn)換后圖像輸入所述第二判別網(wǎng)絡(luò)后得到的值，所述表示將所述D_T(G_T(x))去掉最后一層得到的所述第二特征信息集合，D_S(G_S(x))表示將所述第一轉(zhuǎn)換后圖像輸入所述第一判別網(wǎng)絡(luò)后得到的值，所述表示將所述D_S(G_S(x))去掉最后一層后得到的所述第一特征信息集合，所述表示所述第二特征信息集合與所述第一特征信息集合的差值的一范數(shù)，所述表示所述的平方，所述L_prec(G_S)表示所述第二損失函數(shù)。