[發明專利]風格圖像生成方法、模型訓練方法、裝置、設備和介質

有效

• 申請號: 202011063185.2
• 申請日: 2020-09-30
• 公開（公告）號: CN112989904B
• 公開（公告）日: 2022-03-25
• 發明（設計）人: 胡興鴻;尹淳驥
• 申請（專利權）人: 北京字節跳動網絡技術有限公司
• 主分類號: G06V40/16
• 分類號: G06V40/16;G06V10/774;G06K9/62;G06T5/50;G06T7/11
• 代理公司: 北京開陽星知識產權代理有限公司 11710
• 代理人: 安偉

本公開實施例涉及一種風格圖像生成方法、模型訓練方法、裝置、設備和介質，其中，風格圖像生成方法包括：獲取原始人臉圖像；利用預先訓練的風格圖像生成模型，得到與原始人臉圖像對應的目標風格人臉圖像；其中，風格圖像生成模型基于多個原始人臉樣本圖像和多個目標風格人臉樣本圖像訓練得到，多個目標風格人臉樣本圖像由預先訓練的圖像生成模型生成，并且，圖像生成模型基于預先獲取的多個標準風格人臉樣本圖像訓練得到。本公開實施例可以解決現有方案中圖像風格轉換后圖像效果不佳的問題，提高風格圖像的生成效果。

技術領域

本公開涉及圖像處理技術領域，尤其涉及一種風格圖像生成方法、模型訓練方法、裝置、設備和介質。

背景技術

目前，隨著視頻交互應用的功能逐漸豐富化，圖像風格轉換成為了一種新的趣味性玩法。圖像風格轉換是指將一幅或者多幅圖像進行風格轉換，生成符合用戶需求的風格圖像。

然而，在現有技術中，當對圖像進行風格轉換時，往往轉換后的圖像效果并不理想。以人臉圖像為例，考慮拍照角度、拍照方式的差異，導致不同的原始人臉圖像的構圖方式、圖像尺寸等存在差異，并且，具有風格圖像生成功能的模型的訓練效果也參差不齊，進而基于訓練得到的模型對這些存在差異的人臉圖像進行風格轉換過程中，風格轉換后的圖像效果并不理想。

發明內容

為了解決上述技術問題或者至少部分地解決上述技術問題，本公開實施例提供了一種風格圖像生成方法、模型訓練方法、裝置、設備和介質。

第一方面，本公開實施例提供了一種風格圖像生成方法，包括：

獲取原始人臉圖像；

利用預先訓練的風格圖像生成模型，得到與所述原始人臉圖像對應的目標風格人臉圖像；

其中，所述風格圖像生成模型基于多個原始人臉樣本圖像和多個目標風格人臉樣本圖像訓練得到，所述多個目標風格人臉樣本圖像由預先訓練的圖像生成模型生成，并且，所述圖像生成模型基于預先獲取的多個標準風格人臉樣本圖像訓練得到。

第二方面，本公開實施例還提供了一種風格圖像生成模型的訓練方法，包括：

獲取多個原始人臉樣本圖像；

獲取多個標準風格人臉樣本圖像；

基于所述多個標準風格人臉樣本圖像，對圖像生成模型進行訓練，獲得訓練后的圖像生成模型；

基于所述訓練后的圖像生成模型生成多個目標風格人臉樣本圖像；

利用所述多個原始人臉樣本圖像和所述多個目標風格人臉樣本圖像對風格圖像生成模型進行訓練，獲得訓練后的風格圖像生成模型。

第三方面，本公開實施例還提供了一種風格圖像生成裝置，包括：

原始圖像獲取模塊，用于獲取原始人臉圖像；

風格圖像生成模塊，用于利用預先訓練的風格圖像生成模型，得到與所述原始人臉圖像對應的目標風格人臉圖像；

其中，所述風格圖像生成模型基于多個原始人臉樣本圖像和多個目標風格人臉樣本圖像訓練得到，所述多個目標風格人臉樣本圖像由預先訓練的圖像生成模型生成，并且，所述圖像生成模型基于預先獲取的多個標準風格人臉樣本圖像訓練得到。

第四方面，本公開實施例還提供了一種風格圖像生成模型的訓練裝置，包括：

原始樣本圖像獲取模塊，用于獲取多個原始人臉樣本圖像；

圖像生成模型訓練模塊，用于獲取多個標準風格人臉樣本圖像，并基于所述多個標準風格人臉樣本圖像，對圖像生成模型進行訓練，獲得訓練后的圖像生成模型；

目標風格樣本圖像生成模塊，用于基于所述訓練后的圖像生成模型生成多個目標風格人臉樣本圖像；