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

在審

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

本公開的實施例公開了圖像處理方法、胡子變換網絡的訓練方法、裝置、電子設備和計算機可讀介質。該圖像處理方法的一具體實施方式包括：獲取目標圖像；將該目標圖像輸入至預先訓練的胡子變換網絡，輸出變換胡子后的圖像，其中，該預先訓練的胡子變換網絡的訓練樣本集是基于預先設定的類別比例獲取的，該預先訓練的胡子變換網絡對應的訓練批次是基于預先設定的采樣比劃分的。該實施方式可以有效、準確的對該目標圖像包括的目標生物體進行變換胡子。

技術領域

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

背景技術

隨著計算機視覺技術的發展，計算機視覺技術應用于更為廣泛的領域。進而，計算機視覺技術可以應用在變換胡子方面以增加趣味性。目前，大多技術采用目標檢測網絡或圖像分割網絡來實現基本的胡子變換。但是，在對上述目標檢測網絡或圖像分割網絡訓練時，訓練樣本集的選取和批次的劃分不夠合理，導致不能有效、準確的對目標圖像中的目標對象進行變換胡子。

發明內容

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

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

第一方面，本公開的一些實施例提供了一種圖像處理方法，該方法包括：獲取目標圖像；將上述目標圖像輸入至預先訓練的胡子變換網絡，輸出變換胡子后的圖像，其中，上述預先訓練的胡子變換網絡的訓練樣本集是基于預先設定的類別比例獲取的，上述預先訓練的胡子變換網絡對應的訓練批次是基于預先設定的采樣比劃分的。

第二方面，本公開的一些實施例提供了一種胡子變換網絡的訓練方法，該方法包括：基于預先設定的類別比例，獲取圖像集合；基于預先設定的采樣比和上述圖像集合，生成預定數目個批次；基于胡子區域與胡子所在臉部區域的面積比例，調整初始胡子變換網絡的損失函數，得到調整后的胡子變換網絡；利用上述預定數目個批次，對上述調整后的胡子變換網絡進行訓練，得到訓練后的胡子變換網絡。

第三方面，本公開的一些實施例提供了一種圖像處理裝置，裝置包括：獲取單元，被配置成獲取目標圖像；生成單元，被配置成將上述目標圖像輸入至預先訓練的胡子變換網絡，輸出變換胡子后的圖像，其中，上述預先訓練的胡子變換網絡的訓練樣本集是基于預先設定的類別比例獲取的，上述預先訓練的胡子變換網絡對應的訓練批次是基于預先設定的采樣比劃分的。

第四方面，本公開的一些實施例提供了一種胡子變換網絡的訓練訓練裝置，裝置包括：獲取單元，被配置成基于預先設定的類別比例，獲取圖像集合；生成單元，被配置成基于預先設定的采樣比和上述圖像集合，生成預定數目個批次；調整單元，被配置成基于胡子區域與胡子所在臉部區域的面積比例，調整初始胡子變換網絡的損失函數，得到調整后的胡子變換網絡；訓練單元，被配置成利用上述預定數目個批次，對上述調整后的胡子變換網絡進行訓練，得到訓練后的胡子變換網絡。

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

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

本公開的上述各個實施例中的多個實施例具有如下有益效果：首先，獲取目標圖像。然后，將上述目標圖像輸入至預先訓練的胡子變換網絡，可以有效、準確的輸出變換胡子后的圖像。其中，上述預先訓練的胡子變換網絡的訓練樣本集是基于預先設定的類別比例獲取的，上述預先訓練的胡子變換網絡對應的訓練批次是基于預先設定的采樣比劃分的。