[發明專利]圖像處理方法、胡子變換網絡的訓練方法、裝置

在審

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

1.一種圖像處理方法，包括：

獲取目標圖像；

將所述目標圖像輸入至預先訓練的胡子變換網絡，輸出變換胡子后的圖像，其中，所述預先訓練的胡子變換網絡的訓練樣本集是基于預先設定的類別比例獲取的，所述預先訓練的胡子變換網絡對應的訓練批次是基于預先設定的采樣比劃分的。

2.根據權利要求1所述的方法，其中，所述訓練樣本集通過以下步驟生成：

基于所述類別比例，獲取圖像集合；

對所述圖像集合中的每張圖像進行目標生物體臉部檢測，得到臉部區域集合；

提取所述臉部區域集合中每個臉部區域的預定數目個第一臉部關鍵點作為第一臉部關鍵點集合；

基于預先獲取的臉部圖像和所述第一臉部關鍵點集合，對所述臉部區域進行校正，生成校正后的臉部區域；

基于所得到的校正后的臉部區域，生成處理后的圖像；

基于所述處理后的圖像，生成訓練樣本。

3.根據權利要求2所述的方法，其中，所述基于預先獲取的臉部圖像和所述第一臉部關鍵點集合，對所述臉部區域進行校正，生成校正后的臉部區域，包括：

提取所述臉部圖像的所述預定數目個第二臉部關鍵點作為第二臉部關鍵點集合；

確定所述第一臉部關鍵點集合與所述第二臉部關鍵點集合之間的仿射變換矩陣；

利用所述仿射變換矩陣，對所述臉部區域進行校正，生成校正后的臉部區域。

4.根據權利要求1所述的方法，其中，所述胡子變換網絡為去胡子網絡；以及

所述將所述目標圖像輸入至預先訓練的胡子變換網絡，輸出變換胡子后的圖像，包括：

響應于對所述目標圖像進行去胡子特效，將所述目標圖像輸入至預先訓練的去胡子網絡，輸出去胡子后的目標圖像，其中，所述去胡子網絡包括胡子分割網絡。

5.根據權利要求1所述的方法，其中，所述胡子變換網絡為加胡子網絡；以及

所述將所述目標圖像輸入至預先訓練的胡子變換網絡，輸出變換胡子后的圖像，包括：

響應于對所述目標圖像進行加胡子特效，將所述目標圖像輸入至預先訓練的加胡子網絡，輸出添加胡子后的目標圖像，其中，所述加胡子網絡包括胡子檢測網絡。

6.根據權利要求2所述的方法，其中，所述臉部圖像是臉部特征滿足預定條件的圖像。

7.根據權利要求1所述的方法，其中，所述預先設定的類別比例包括所述訓練樣本集中的男性圖像數目與女性圖像數目的比例和所述訓練樣本集中男性有胡子圖像數目與男性無胡子圖像數目的比例。

8.根據權利要求1所述的方法，其中，所述采樣比是男性有胡子圖像數目、男性無胡子圖像數目、女性有胡子圖像數目和女性無胡子圖像數目之間的比例。

9.一種胡子變換網絡的訓練方法，包括：

基于預先設定的類別比例，獲取圖像集合；

基于預先設定的采樣比和所述圖像集合，生成預定數目個批次；

基于胡子區域與胡子所在臉部區域的面積比例，調整初始胡子變換網絡的損失函數，得到調整后的胡子變換網絡；

利用所述預定數目個批次，對所述調整后的胡子變換網絡進行訓練，得到訓練后的胡子變換網絡。

10.根據權利要求9所述的方法，其中，所述基于預先設定的采樣比和所述圖像集合，生成預定數目個批次，包括：

對所述圖像集合中的每張圖像包括的臉部區域進行校正處理，生成處理后的圖像，得到處理后的圖像集合；

提取所述處理后的圖像集合中每個圖像的紋理特征信息；

將所述紋理特征信息與所述圖像進行疊加，生成疊加后的結果作為疊加后的圖像，得到疊加后的圖像集合作為訓練樣本集；

基于所述采樣比和所述訓練樣本集，生成預定數目個批次。