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

在審

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

本公開的實施例公開了臉部圖像處理模型訓練方法、裝置、電子設備和計算機可讀介質。該方法的一具體實施方式包括：獲取訓練樣本集；從訓練樣本集中選取訓練樣本組，訓練樣本組中顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像的數量的比例符合目標比例，目標比例在訓練開始階段為第一預設值，在訓練結束階段為第二預設值；將訓練樣本組中的臉部圖像作為輸入，將臉部圖像對應的處理后臉部圖像作為期望輸出，訓練得到臉部圖像處理模型。該實施方式實現了提高了對于臉部圖像中的局部區域的處理質量。

技術領域

本公開的實施例涉及計算機技術領域，具體涉及臉部圖像處理模型訓練方法、裝置、電子設備和計算機可讀介質。

背景技術

圖像處理技術通過對圖像進行不同的加工以滿足不同的需求。例如，可以對破損或模糊的圖像進行修復，從而得到經過處理的清晰完整的圖像。隨著人工智能技術的發展，人工神經網絡在圖像處理領域發揮了重要作用。但是，人工神經網絡在一些場景下存在提高局部區域處理質量的需求。例如，在對包含人臉區域的破損圖像進行修復時，局部區域(例如牙齒區域)的處理質量不高。

發明內容

本公開的內容部分用于以簡要的形式介紹構思，這些構思將在后面的具體實施方式部分被詳細描述。本公開的內容部分并不旨在標識要求保護的技術方案的關鍵特征或必要特征，也不旨在用于限制所要求的保護的技術方案的范圍。

本公開的一些實施例提出了臉部圖像處理模型訓練方法、裝置、電子設備和計算機可讀介質，來解決以上背景技術部分提到的技術問題。

第一方面，本公開的一些實施例提供了一種臉部圖像處理模型訓練方法，包括：獲取訓練樣本集，其中，訓練樣本集中的訓練樣本包括臉部圖像和臉部圖像對應的處理后臉部圖像，臉部圖像包括顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像；從訓練樣本集中選取訓練樣本組，訓練樣本組中顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像的數量的比例符合目標比例，目標比例在訓練開始階段為第一預設值，在訓練結束階段為第二預設值；將訓練樣本組中的臉部圖像作為輸入，將臉部圖像對應的處理后臉部圖像作為期望輸出，訓練得到臉部圖像處理模型。

第二方面，本公開的一些實施例提供了一種臉部圖像處理模型訓練裝置，包括：獲取單元，被配置成獲取訓練樣本集，其中，訓練樣本集中的訓練樣本包括臉部圖像和臉部圖像對應的處理后臉部圖像，臉部圖像包括顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像；選取單元，被配置成從訓練樣本集中選取訓練樣本組，訓練樣本組中顯示目標部位的臉部圖像和未顯示目標部位的臉部圖像的數量的比例符合目標比例，目標比例在訓練開始階段為第一預設值，在訓練結束階段為第二預設值；訓練單元，被配置成將訓練樣本組中的臉部圖像作為輸入，將臉部圖像對應的處理后臉部圖像作為期望輸出，訓練得到臉部圖像處理模型。

第三方面，本公開的一些實施例提供了一種電子設備，包括：一個或多個處理器；存儲裝置，其上存儲有一個或多個程序，當一個或多個程序被一個或多個處理器執行，使得一個或多個處理器實現如第一方面中任一實現方式描述的方法。

第四方面，本公開的一些實施例提供了一種計算機可讀介質，其上存儲有計算機程序，其中，程序被處理器執行時實現如第一方面中任一實現方式描述的方法。