[發明專利]發型遷移模型訓練方法、發型遷移方法及相關裝置

在審

• 申請號: 202110409682.1
• 申請日: 2021-04-16
• 公開（公告）號: CN113191404A
• 公開（公告）日: 2021-07-30
• 發明（設計）人: 陳仿雄
• 申請（專利權）人: 深圳數聯天下智能科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/46;G06N3/04;G06N3/08
• 代理公司: 暫無信息
• 代理人: 暫無信息

本申請提供發型遷移模型訓練方法、發型遷移方法及相關裝置，其中，發型遷移模型訓練方法包括：獲取樣本發型圖像和所樣本參考發型圖像；對樣本發型圖像和樣本參考發型圖像進行特征提取和特征融合，得到樣本融合特征圖像；對樣本融合特征圖像進行特征還原，以得到樣本遷移發型圖像；將樣本遷移發型圖像輸入至圖像判別模型和發型預測模型，以得到判別結果和發型預測結果；基于樣本遷移發型圖像和樣本發型圖像，以及，發型預測結果和樣本參考發型圖像中的發型，以及判別結果，對發型遷移模型、圖像判別模型以及發型預測模型進行迭代調參，直至發型遷移模型、圖像判別模型以及發型預測模型均收斂。該技術方案可以使虛擬美發的融合效果更自然。

技術領域

本申請涉及圖像處理領域，尤其涉及發型遷移模型訓練方法、發型遷移方法及相關裝置。

背景技術

美容美發是在用戶行為中一種目標較為常見的行為，在進行美容美發之前，許多用戶希望看到自己美容美發后的形象，因此，一些基于圖像處理技術的虛擬美發應用應運而生。

在目前的虛擬美發應用中，在獲取到待美發圖像后，通過分割算法先從待美發圖像中分割出頭發區域，然后將模板中的模板發型遷移至頭發區域，即用美發模板中的模板發型替換頭發區域中的發型，并將模板發型與待美發圖像中除頭發區域以外的其他區域進行融合，得到美發后的效果圖。這種方式過度依賴模板，融合效果較生硬。

發明內容

本申請提供發型遷移模型訓練方法、發型遷移方法及相關裝置，以解決現有虛擬美發技術融合效果較生硬的問題。

第一方面，提供一種發型遷移模型訓練方法，其中，發型遷移模型包括特征提取模塊、特征融合模塊以及特征還原模塊，該方法包括如下步驟：

獲取樣本發型圖像和上述樣本發型圖像對應的樣本參考發型圖像，其中，上述樣本參考發型圖像中的發型不同于上述樣本發型圖像中的發型；

通過上述特征提取模塊分別對上述樣本發型圖像和上述樣本參考發型圖像進行特征提取，以得到第一樣本特征圖像和第二樣本特征圖像；

通過上述特征融合模塊將上述第一樣本特征圖像與上述第二樣本特征圖像進行特征融合，得到樣本融合特征圖像；

通過上述特征還原模塊對上述樣本融合特征圖像進行特征還原，以得到上述樣本發型圖像對應的樣本遷移發型圖像；

將上述樣本遷移發型圖像輸入至圖像判別模型，以得到上述樣本遷移發型圖像的判別結果，上述判別結果用于指示上述樣本遷移發型圖像為真實樣本發型圖像的可能性；

將上述樣本遷移發型圖像輸入至發型預測模型，以得到上述樣本遷移發型圖像的發型預測結果；

基于上述樣本遷移發型圖像和上述樣本發型圖像，以及，上述發型預測結果和上述樣本參考發型圖像中的發型，以及上述判別結果，對上述發型遷移模型、上述圖像判別模型以及上述發型預測模型進行迭代調參，直至上述發型遷移模型、上述圖像判別模型以及上述發型預測模型均收斂；

將收斂后的發型遷移模型確定為目標發型遷移模型。