[發(fā)明專利]模型訓(xùn)練方法、裝置和存儲介質(zhì)有效
| 申請?zhí)枺?/td> | 201910833116.6 | 申請日： | 2018-01-26 |
| 公開（公告）號： | CN110555821B | 公開（公告）日： | 2022-04-05 |
| 發(fā)明（設(shè)計）人： | 李旻駿;黃浩智;馬林;劉威 | 申請（專利權(quán)）人： | 騰訊科技（深圳）有限公司 |
| 主分類號： | G06T5/50 | 分類號： | G06T5/50;G06T3/00;G06N3/08;G06N3/04 |
| 代理公司： | 華進聯(lián)合專利商標(biāo)代理有限公司 44224 | 代理人： | 楊歡 |
| 地址： | 518000 廣東省深圳*** | 國省代碼： | 廣東;44 |
| 權(quán)利要求書： | 查看更多 | 說明書： | 查看更多 |
| 摘要： | |||
| 搜索關(guān)鍵詞： | 模型 訓(xùn)練 方法 裝置 存儲 介質(zhì) | ||
1.一種模型訓(xùn)練方法，包括：
獲取屬于第一圖像類別的第一樣本和屬于第二圖像類別的第二樣本；
將所述第一樣本依次經(jīng)過第一階段圖像轉(zhuǎn)化模型和第一階段圖像逆向轉(zhuǎn)化模型，得到第一樣本單階段恢復(fù)圖像；
將所述第二樣本依次經(jīng)過所述第一階段圖像逆向轉(zhuǎn)化模型和所述第一階段圖像轉(zhuǎn)化模型，得到第二樣本單階段恢復(fù)圖像；
按照所述第一樣本與所述第一樣本單階段恢復(fù)圖像的差異，及所述第二樣本與所述第二樣本單階段恢復(fù)圖像的差異，調(diào)整所述第一階段圖像轉(zhuǎn)化模型和所述第一階段圖像逆向轉(zhuǎn)化模型，直至滿足訓(xùn)練停止條件時結(jié)束訓(xùn)練；
將所述第一樣本依次經(jīng)過所述第一階段圖像轉(zhuǎn)化模型和第二階段圖像轉(zhuǎn)化模型，根據(jù)所述第一階段圖像轉(zhuǎn)化模型輸出的第一樣本一階轉(zhuǎn)化圖像，和所述第二階段圖像轉(zhuǎn)化模型輸出的第一樣本二階轉(zhuǎn)化圖像，得到與所述第一樣本對應(yīng)、且屬于所述第二圖像類別的第一樣本轉(zhuǎn)化圖像；
將所述第一樣本轉(zhuǎn)化圖像依次經(jīng)過所述第一階段圖像逆向轉(zhuǎn)化模型和第二階段圖像逆向轉(zhuǎn)化模型，根據(jù)所述第一階段圖像逆向轉(zhuǎn)化模型輸出的第一樣本一階恢復(fù)圖像，和所述第二階段圖像逆向轉(zhuǎn)化模型輸出的第一樣本二階恢復(fù)圖像，得到與所述第一樣本對應(yīng)、且屬于所述第一圖像類別的第一樣本恢復(fù)圖像；
將所述第二樣本依次經(jīng)過所述第一階段圖像逆向轉(zhuǎn)化模型和第二階段圖像逆向轉(zhuǎn)化模型，根據(jù)所述第一階段圖像逆向轉(zhuǎn)化模型輸出的第二樣本一階轉(zhuǎn)化圖像，和所述第二階段圖像逆向轉(zhuǎn)化模型輸出的第二樣本二階轉(zhuǎn)化圖像，得到與所述第二樣本對應(yīng)、且屬于所述第一圖像類別的第二樣本轉(zhuǎn)化圖像；
將所述第二樣本轉(zhuǎn)化圖像依次經(jīng)過所述第一階段圖像轉(zhuǎn)化模型和第二階段圖像轉(zhuǎn)化模型，根據(jù)所述第一階段圖像轉(zhuǎn)化模型輸出的第二樣本一階恢復(fù)圖像，和所述第二階段圖像轉(zhuǎn)化模型輸出的第二樣本二階恢復(fù)圖像，得到與所述第二樣本對應(yīng)、且屬于所述第二圖像類別的第二樣本恢復(fù)圖像；
按照所述第一樣本與所述第一樣本恢復(fù)圖像的差異，及所述第二樣本與所述第二樣本恢復(fù)圖像的差異，調(diào)整所述第二階段圖像轉(zhuǎn)化模型和所述第二階段圖像逆向轉(zhuǎn)化模型，直至滿足訓(xùn)練停止條件時結(jié)束訓(xùn)練。
2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述方法還包括：
獲取所述第一樣本經(jīng)過所述第一階段圖像轉(zhuǎn)化模型后，由所述第一階段圖像轉(zhuǎn)化模型輸出的第一樣本單階段轉(zhuǎn)化圖像；
獲取所述第二樣本經(jīng)過所述第一階段圖像逆向轉(zhuǎn)化模型后，由所述第一階段圖像逆向轉(zhuǎn)化模型輸出的第二樣本單階段轉(zhuǎn)化圖像；
將所述第一樣本和所述第二樣本單階段轉(zhuǎn)化圖像分別輸入第一階段圖像逆向轉(zhuǎn)化鑒別模型，分別得到所述第一樣本的鑒別置信度和所述第二樣本單階段轉(zhuǎn)化圖像的鑒別置信度；
將所述第二樣本和所述第一樣本單階段轉(zhuǎn)化圖像分別輸入第一階段圖像轉(zhuǎn)化鑒別模型，分別得到所述第二樣本的鑒別置信度和所述第一樣本單階段轉(zhuǎn)化圖像的鑒別置信度；
所述按照所述第一樣本與所述第一樣本單階段恢復(fù)圖像的差異，及所述第二樣本與所述第二樣本單階段恢復(fù)圖像的差異，調(diào)整所述第一階段圖像轉(zhuǎn)化模型和所述第一階段圖像逆向轉(zhuǎn)化模型，直至滿足訓(xùn)練停止條件時結(jié)束訓(xùn)練，包括：
按照最大化所述第一樣本的鑒別置信度和所述第二樣本的鑒別置信度的方向、最小化所述第二樣本單階段轉(zhuǎn)化圖像的鑒別置信度、所述第一樣本單階段轉(zhuǎn)化圖像的鑒別置信度、所述第一樣本與所述第一樣本單階段恢復(fù)圖像的差異，及所述第二樣本與所述第二樣本單階段恢復(fù)圖像的差異的方向，調(diào)整所述第一階段圖像轉(zhuǎn)化鑒別模型、所述第一階段圖像逆向轉(zhuǎn)化鑒別模型、所述第一階段圖像轉(zhuǎn)化模型和所述第一階段圖像逆向轉(zhuǎn)化模型，直至滿足訓(xùn)練停止條件時結(jié)束訓(xùn)練。
該專利技術(shù)資料僅供研究查看技術(shù)是否侵權(quán)等信息，商用須獲得專利權(quán)人授權(quán)。該專利全部權(quán)利屬于騰訊科技（深圳）有限公司，未經(jīng)騰訊科技（深圳）有限公司許可，擅自商用是侵權(quán)行為。如果您想購買此專利、獲得商業(yè)授權(quán)和技術(shù)合作，請聯(lián)系【客服】
本文鏈接：http://www.szxzyx.cn/pat/books/201910833116.6/1.html，轉(zhuǎn)載請聲明來源鉆瓜專利網(wǎng)。
