[發明專利]人臉關鍵點檢測模型的訓練方法、裝置、計算機設備及存儲介質

在審

• 申請號: 202011528500.4
• 申請日: 2020-12-22
• 公開（公告）號: CN112598807A
• 公開（公告）日: 2021-04-02
• 發明（設計）人: 馬嘯
• 申請（專利權）人: 深圳數聯天下智能科技有限公司
• 主分類號: G06T19/20
• 分類號: G06T19/20;G06K9/00;G06K9/62
• 代理公司: 深圳中細軟知識產權代理有限公司 44528
• 代理人: 王志強

本發明公開了一種人臉關鍵點檢測模型的訓練方法，包括：獲取三維人臉模型，確定三維人臉模型在初始位置對應的人臉關鍵點標注；確定三維人臉模型在三維空間中的移動路徑；對三維人臉模型進行拍照得到新的二維人臉圖像；基于拍照時三維人臉模型在移動路徑中的位置和在初始位置對應的人臉關鍵點標注計算得到新的二維人臉圖像對應的人臉關鍵點標注；將新的二維人臉圖像和新的二維人臉圖像對應的人臉關鍵點標注加入到訓練數據集，并利用訓練數據集對人臉關鍵點檢測模型進行訓練。基于訓練數據集訓練得到的人臉關鍵點檢測模型能夠準確且穩定地對視頻流中的人臉關鍵點進行識別。此外，還提出了一種人臉關鍵點檢測模型的訓練裝置、計算機設備及存儲介質。

技術領域

本發明涉及計算機技術領域，具體涉及一種人臉關鍵點檢測模型的訓練方法、裝置、計算機設備及存儲介質。

背景技術

人臉關鍵點的作用是精確定位和分割人臉各部位的位置，比如眼睛、眉毛，嘴巴的精確外輪廓，臉型的外輪廓等。主要應用包括人臉變形(瘦臉、大眼等)、虛擬裝飾、虛擬試妝和動畫電影等領域。

人臉關鍵點檢測模型檢測的精準度依賴于訓練數據集，傳統的人臉關鍵點的訓練數據集是隨機收集不同的人臉照片，并在照片上手工進行關鍵點標注。由于人臉關鍵點的定義并不是精確唯一的，手工標注的人臉關鍵點是在一定范圍內振動的。

對于視頻流，人臉關鍵點的檢測不但需要高準確性，而且要求高穩定性，最理想的狀態是，相鄰幀檢測到的同一關鍵點要在物理空間上的同一位置，否則，會出現關鍵點抖動的視覺效果。

但是，現階段幾乎所有模型的訓練數據都是單人單張圖像的手工標注，沒有考慮連續視頻幀之間的位置關系，使得同一關鍵點在不同視頻幀上的位置有偏差，在連續的視頻流中顯示的效果就是人臉關鍵點抖動。

發明內容

基于此，有必要針對上述問題，提出一種人臉關鍵點檢測模型的訓練方法、裝置、計算機設備及存儲介質。通過該方式得到的訓練數據集包含有對連續視頻幀的標注，基于該訓練數據集訓練得到的人臉關鍵點檢測模型能夠準確且穩定地對視頻流中的人臉關鍵點進行識別。

一種人臉關鍵點檢測模型的訓練方法，包括：

獲取三維人臉模型，確定所述三維人臉模型在初始位置對應的人臉關鍵點標注，所述人臉關鍵點標注用于指示所述人臉關鍵點的位置坐標；

確定所述三維人臉模型在三維空間中的移動路徑，以使所述三維人臉模型按照所述移動路徑進行移動；

在所述三維人臉模型按照所述移動路徑移動的過程中，對所述三維人臉模型進行拍照得到新的二維人臉圖像；

基于拍照時所述三維人臉模型在所述移動路徑中的位置和所述在初始位置對應的人臉關鍵點標注計算得到所述新的二維人臉圖像對應的人臉關鍵點標注；

將所述新的二維人臉圖像和所述新的二維人臉圖像對應的人臉關鍵點標注加入到所述訓練數據集，并利用所述訓練數據集對人臉關鍵點檢測模型進行訓練。

一種人臉關鍵點檢測模型的訓練裝置，包括：

獲取三維人臉模型，確定所述三維人臉模型在初始位置對應的人臉關鍵點標注，所述人臉關鍵點標注用于指示所述人臉關鍵點的位置坐標；

確定所述三維人臉模型在三維空間中的移動路徑，以使所述三維人臉模型按照所述移動路徑進行移動；

在所述三維人臉模型按照所述移動路徑移動的過程中，對所述三維人臉模型進行拍照得到新的二維人臉圖像；

基于拍照時所述三維人臉模型在所述移動路徑中的位置和所述在初始位置對應的人臉關鍵點標注計算得到所述新的二維人臉圖像對應的人臉關鍵點標注；將所述新的二維人臉圖像和所述新的二維人臉圖像對應的人臉關鍵點標注加入到所述訓練數據集，并利用所述訓練數據集對人臉關鍵點檢測模型進行訓練。