[發明專利]圖像生成方法及裝置

有效

• 申請號: 201711123279.2
• 申請日: 2017-11-14
• 公開（公告）號: CN107845062B
• 公開（公告）日: 2021-08-17
• 發明（設計）人: 張水發
• 申請（專利權）人: 北京小米移動軟件有限公司
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06Q30/06
• 代理公司: 北京英創嘉友知識產權代理事務所(普通合伙) 11447
• 代理人: 魏嘉熹

1.一種圖像生成方法，其特征在于，包括：

根據未戴眼鏡人臉圖像和對應的戴眼鏡人臉圖像，訓練得到圖像生成網絡；

將待處理的未戴眼鏡人臉圖像輸入所述圖像生成網絡，生成第一試戴眼鏡人臉圖像；

根據未戴眼鏡人臉圖像和對應的戴眼鏡人臉圖像，訓練得到圖像生成網絡，包括：

根據損失函數、所述未戴眼鏡人臉圖像和所述對應的戴眼鏡人臉圖像，訓練判別網絡和生成網絡，所述判別網絡用于判斷輸入的圖像與所述對應的戴眼鏡人臉圖像的一致性；

重復訓練所述生成網絡和所述判別網絡，在訓練次數達到預設閾值或所述判別網絡輸出的判別結果處于收斂時，將當前的生成網絡確定為所述圖像生成網絡；

其中，所述損失函數由所述生成網絡、所述判別網絡以及圖像的像素點的權重確定；所述像素點的權重由所述像素點與目標像素點的距離確定，所述目標像素點為圖像中眼睛區域的像素點。

2.根據權利要求1所述的方法，其特征在于，根據損失函數、所述未戴眼鏡人臉圖像和所述對應的戴眼鏡人臉圖像，訓練判別網絡和生成網絡，包括：

將所述未戴眼鏡人臉圖像輸入到生成網絡中，生成第二試戴眼鏡人臉圖像；

將所述第二試戴眼鏡人臉圖像輸入到判別網絡中，得到第一判別結果；

將所述對應的戴眼鏡人臉圖像輸入到判別網絡中，得到第二判別結果；

計算所述對應的戴眼鏡人臉圖像與所述第二試戴眼鏡人臉圖像之間的距離損失結果；

根據所述第一判別結果、所述第二判別結果、所述距離損失結果以及所述圖像的像素點的權重，通過所述損失函數計算得到損失；

根據所述損失，調整所述判別網絡或生成網絡中的參數的權重。

3.一種圖像生成裝置，其特征在于，包括：

訓練模塊，用于根據未戴眼鏡人臉圖像和對應的戴眼鏡人臉圖像，訓練得到圖像生成網絡；

生成模塊，用于將待處理的未戴眼鏡人臉圖像輸入所述圖像生成網絡，生成第一試戴眼鏡人臉圖像；

所述訓練模塊包括：

訓練子模塊，用于根據損失函數、所述未戴眼鏡人臉圖像和所述對應的戴眼鏡人臉圖像，訓練判別網絡和生成網絡，所述判別網絡用于判斷輸入的圖像與所述對應的戴眼鏡人臉圖像的一致性；

確定子模塊，用于重復訓練所述生成網絡和所述判別網絡，在訓練次數達到預設閾值或所述判別網絡輸出的判別結果處于收斂時，將當前的生成網絡確定為所述圖像生成網絡；

其中，所述損失函數由所述生成網絡、所述判別網絡以及圖像的像素點的權重確定；所述像素點的權重由所述像素點與目標像素點的距離確定，所述目標像素點為圖像中眼睛區域的像素點。

4.根據權利要求3所述的裝置，其特征在于，所述訓練子模塊還用于：

將所述未戴眼鏡人臉圖像輸入到生成網絡中，生成第二試戴眼鏡人臉圖像；

將所述第二試戴眼鏡人臉圖像輸入到判別網絡中，得到第一判別結果；

將所述對應的戴眼鏡人臉圖像輸入到判別網絡中，得到第二判別結果；

計算所述對應的戴眼鏡人臉圖像與所述第二試戴眼鏡人臉圖像之間的距離損失結果；

根據所述第一判別結果、所述第二判別結果、所述距離損失結果以及所述圖像的像素點的權重，通過所述損失函數計算得到損失；

根據所述損失，調整所述判別網絡或生成網絡中的參數的權重。

5.一種圖像生成裝置，其特征在于，包括：

處理器；

用于存儲處理器可執行指令的存儲器；

其中，所述處理器被配置為：執行權利要求1或2所述的方法。

6.一種非臨時性計算機可讀存儲介質，其特征在于，當所述存儲介質中的指令由處理器執行時，使得處理器能夠執行根據權利要求1或2所述的方法。