[發(fā)明專利]圖像生成方法及裝置

有效

• 申請?zhí)枺?/td>: 201711123279.2
• 申請日: 2017-11-14
• 公開（公告）號: CN107845062B
• 公開（公告）日: 2021-08-17
• 發(fā)明（設(shè)計）人: 張水發(fā)
• 申請（專利權(quán)）人: 北京小米移動軟件有限公司
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06Q30/06
• 代理公司: 北京英創(chuàng)嘉友知識產(chǎn)權(quán)代理事務(wù)所(普通合伙) 11447
• 代理人: 魏嘉熹

本公開是關(guān)于一種圖像生成方法及裝置。該方法包括：根據(jù)未戴眼鏡人臉圖像和對應(yīng)的戴眼鏡人臉圖像，訓(xùn)練得到圖像生成網(wǎng)絡(luò)；將待處理的未戴眼鏡人臉圖像輸入圖像生成網(wǎng)絡(luò)，生成第一試戴眼鏡人臉圖像。根據(jù)本公開的實施例，通過未戴眼鏡人臉圖像與對應(yīng)戴眼鏡人臉圖像訓(xùn)練得到圖像生成網(wǎng)絡(luò)，從而實現(xiàn)將待處理的未戴眼鏡人臉圖像輸入該圖像生成網(wǎng)絡(luò)時，可以生成具有較高真實度的第一試戴眼鏡人臉圖像。

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

本公開涉及計算機(jī)技術(shù)領(lǐng)域，尤其涉及一種圖像生成方法及裝置。

背景技術(shù)

目前，終端設(shè)備(例如，智能手機(jī)等)上可以安裝多種具有娛樂性質(zhì)的圖像應(yīng)用，這些圖像應(yīng)用可以對圖像中的人臉進(jìn)行娛樂性處理，例如，試戴配飾(例如，試戴眼鏡、試戴帽子)等。然而，現(xiàn)有的圖像應(yīng)用在試戴配飾過程中，具有試戴效果較差、試戴效果不真實等缺點。

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

為克服相關(guān)技術(shù)中存在的問題，本公開提供一種圖像生成方法及裝置。

根據(jù)本公開實施例的第一方面，提供一種圖像生成方法，包括：

根據(jù)未戴眼鏡人臉圖像和對應(yīng)的戴眼鏡人臉圖像，訓(xùn)練得到圖像生成網(wǎng)絡(luò)；

將待處理的未戴眼鏡人臉圖像輸入所述圖像生成網(wǎng)絡(luò)，生成第一試戴眼鏡人臉圖像。

對于以上方法，在一種可能的實現(xiàn)方式中，根據(jù)未戴眼鏡人臉圖像和對應(yīng)的戴眼鏡人臉圖像，訓(xùn)練得到圖像生成網(wǎng)絡(luò)，包括：

根據(jù)損失函數(shù)、所述未戴眼鏡人臉圖像和所述對應(yīng)的戴眼鏡人臉圖像，訓(xùn)練判別網(wǎng)絡(luò)和生成網(wǎng)絡(luò)，所述判別網(wǎng)絡(luò)用于判斷輸入的圖像與所述對應(yīng)的戴眼鏡人臉圖像的一致性；

重復(fù)訓(xùn)練所述生成網(wǎng)絡(luò)和所述判別網(wǎng)絡(luò)，在訓(xùn)練次數(shù)達(dá)到預(yù)設(shè)閾值或所述判別網(wǎng)絡(luò)輸出的判別結(jié)果處于收斂時，將當(dāng)前的生成網(wǎng)絡(luò)確定為所述圖像生成網(wǎng)絡(luò)；

其中，所述損失函數(shù)由所述生成網(wǎng)絡(luò)、所述判別網(wǎng)絡(luò)以及圖像的像素點的權(quán)重確定；所述像素點的權(quán)重由所述像素點與目標(biāo)像素點的距離確定。

對于以上方法，在一種可能的實現(xiàn)方式中，根據(jù)損失函數(shù)、所述未戴眼鏡人臉圖像和所述對應(yīng)的戴眼鏡人臉圖像，訓(xùn)練判別網(wǎng)絡(luò)和生成網(wǎng)絡(luò)，包括：

將所述未戴眼鏡人臉圖像輸入到生成網(wǎng)絡(luò)中，生成第二試戴眼鏡人臉圖像；

將所述第二試戴眼鏡人臉圖像輸入到判別網(wǎng)絡(luò)中，得到第一判別結(jié)果；

將所述對應(yīng)的戴眼鏡人臉圖像輸入到判別網(wǎng)絡(luò)中，得到第二判別結(jié)果；

計算所述對應(yīng)的戴眼鏡人臉圖像與所述第二試戴眼鏡人臉圖像之間的距離損失結(jié)果；

根據(jù)所述第一判別結(jié)果、所述第二判別結(jié)果、所述距離損失結(jié)果以及所述圖像的像素點的權(quán)重，通過所述損失函數(shù)計算得到損失；

根據(jù)所述損失，調(diào)整所述判別網(wǎng)絡(luò)或生成網(wǎng)絡(luò)中的參數(shù)的權(quán)重。

根據(jù)本公開實施例的第二方面，提供一種圖像生成裝置，包括：

訓(xùn)練模塊，用于根據(jù)未戴眼鏡人臉圖像和對應(yīng)的戴眼鏡人臉圖像，訓(xùn)練得到圖像生成網(wǎng)絡(luò)；

生成模塊，用于將待處理的未戴眼鏡人臉圖像輸入所述圖像生成網(wǎng)絡(luò)，生成第一試戴眼鏡人臉圖像。

對于以上裝置，在一種可能的實現(xiàn)方式中，所述訓(xùn)練模塊包括：

訓(xùn)練子模塊，用于根據(jù)損失函數(shù)、所述未戴眼鏡人臉圖像和所述對應(yīng)的戴眼鏡人臉圖像，訓(xùn)練判別網(wǎng)絡(luò)和生成網(wǎng)絡(luò)，所述判別網(wǎng)絡(luò)用于判斷輸入的圖像與所述對應(yīng)的戴眼鏡人臉圖像的一致性；

確定子模塊，用于重復(fù)訓(xùn)練所述生成網(wǎng)絡(luò)和所述判別網(wǎng)絡(luò)，在訓(xùn)練次數(shù)達(dá)到預(yù)設(shè)閾值或所述判別網(wǎng)絡(luò)輸出的判別結(jié)果處于收斂時，將當(dāng)前的生成網(wǎng)絡(luò)確定為所述圖像生成網(wǎng)絡(luò)；