[發明專利]臉部圖像處理模型訓練方法、裝置、電子設備和介質

在審

• 申請號: 202011342855.4
• 申請日: 2020-11-25
• 公開（公告）號: CN112465717A
• 公開（公告）日: 2021-03-09
• 發明（設計）人: 李華夏
• 申請（專利權）人: 北京字跳網絡技術有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06K9/00;G06N3/04;G06N3/08
• 代理公司: 北京衛智暢科專利代理事務所(普通合伙) 11557
• 代理人: 陳佳

1.一種臉部圖像處理模型訓練方法，包括：

獲取訓練樣本集，其中，所述訓練樣本集中的訓練樣本包括臉部圖像和所述臉部圖像對應的處理后臉部圖像，所述臉部圖像包括顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像；

從所述訓練樣本集中選取訓練樣本組，所述訓練樣本組中顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像的數量的比例符合目標比例，所述目標比例在訓練開始階段為第一預設值，在訓練結束階段為第二預設值；

將所述訓練樣本組中的臉部圖像作為輸入，將所述臉部圖像對應的處理后臉部圖像作為期望輸出，訓練得到所述臉部圖像處理模型。

2.根據權利要求1所述的方法，其中，所述顯示目標部位的臉部圖像是通過以下步驟生成的：

獲取原始顯示目標部位的臉部圖像；

確定所述原始顯示目標部位的臉部圖像對應的所述目標部位的掩膜圖像；

將所述目標部位的掩膜圖像和所述原始顯示目標部位的臉部圖像疊加，得到所述顯示目標部位的臉部圖像。

3.根據權利要求1所述的方法，其中，所述將所述訓練樣本組中的臉部圖像作為輸入，將所述臉部圖像對應的處理后臉部圖像作為期望輸出，訓練得到所述臉部圖像處理模型，包括：

將所述訓練樣本組中的臉部圖像輸入初始圖像處理模型，得到實際輸出圖像；

基于預設的損失函數，確定所述實際輸出圖像與所述臉部圖像對應的處理后臉部圖像之間的差異，所述預設的損失函數包括特征匹配損失；

基于所述差異，對所述初始圖像處理模型的參數進行調整。

4.根據權利要求3所述的方法，其中，所述特征匹配損失是通過以下步驟得到的：

將所述初始圖像處理模型中的目標層的輸出結果輸入特征提取網絡，得到所述目標層的特征數據；

基于特征數據得到所述特征匹配損失。

5.根據權利要求1所述的方法，其中，在訓練中間階段，所述目標比例是根據所述第一預設值、所述第二預設值和當前迭代次數確定的。

6.一種臉部圖像處理模型訓練裝置，包括：

獲取單元，被配置成獲取訓練樣本集，其中，所述訓練樣本集中的訓練樣本包括臉部圖像和所述臉部圖像對應的處理后臉部圖像，所述臉部圖像包括顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像；

選取單元，被配置成從所述訓練樣本集中選取訓練樣本組，所述訓練樣本組中顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像的數量的比例符合目標比例，所述目標比例在訓練開始階段為第一預設值，在訓練結束階段為第二預設值；

訓練單元，被配置成將所述訓練樣本組中的臉部圖像作為輸入，將所述臉部圖像對應的處理后臉部圖像作為期望輸出，訓練得到所述臉部圖像處理模型。

7.根據權利要求6所述的裝置，其中，所述顯示目標部位的臉部圖像是通過以下步驟生成的：

獲取原始顯示目標部位的臉部圖像；

確定所述原始顯示目標部位的臉部圖像對應的所述目標部位的掩膜圖像；

將所述目標部位的掩膜圖像和所述原始顯示目標部位的臉部圖像疊加，得到所述顯示目標部位的臉部圖像。

8.根據權利要求6所述的裝置，其中，所述訓練單元進一步被配置成：

將所述訓練樣本組中的臉部圖像輸入初始圖像處理模型，得到實際輸出圖像；

基于預設的損失函數，確定所述實際輸出圖像與所述臉部圖像對應的處理后臉部圖像之間的差異，所述預設的損失函數包括特征匹配損失；

基于所述差異，對所述初始圖像處理模型的參數進行調整。

9.一種電子設備，包括：

一個或多個處理器；

存儲裝置，其上存儲有一個或多個程序，

當所述一個或多個程序被所述一個或多個處理器執行，使得所述一個或多個處理器實現如權利要求1-5中任一所述的方法。

10.一種計算機可讀介質，其上存儲有計算機程序，其中，所述程序被處理器執行時實現如權利要求1-5中任一所述的方法。