[發明專利]模型訓練、人頭重建方法，裝置，設備以及存儲介質

有效

• 申請號: 202110366436.2
• 申請日: 2021-04-06
• 公開（公告）號: CN112884889B
• 公開（公告）日: 2022-05-20
• 發明（設計）人: 王迪
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06T17/00
• 分類號: G06T17/00;G06K9/62;G06V10/774;G06V40/16
• 代理公司: 北京英賽嘉華知識產權代理有限責任公司 11204
• 代理人: 王達佐;馬曉亞

本公開提供了模型訓練、人頭重建方法，裝置，設備以及存儲介質，涉及圖像處理領域，具體涉及計算機視覺、增強現實、深度學習技術領域。具體實現方案為：獲取對預先制作的人頭模型進行展開得到的展開紋理圖像；獲取多張樣本二維人臉圖像；根據多張樣本二維人臉圖像以及展開紋理圖像，確定各樣本二維人臉圖像對應的整頭紋理圖像；將各樣本二維人臉圖像作為輸入，將與所輸入的樣本二維人臉圖像對應的整頭紋理圖像作為期望輸出，訓練得到整頭紋理圖像生成模型。本實現方式可以生成整頭的紋理圖像，從而使得人頭模型更真實。

技術領域

本公開涉及圖像處理領域，具體涉及計算機視覺、增強現實、深度學習技術領域，尤其涉及模型訓練、人頭重建方法，裝置，設備以及存儲介質。

背景技術

通過二維人臉圖重建三維人臉是目前成本最低的三維重建方式，越來越多的研究人員使用這一方法進行人物真實效果重建工作。但是單張二維人臉圖只能表現出前臉區域信息，并沒有人頭完整信息，缺失了人頭兩側、后側、頂部的信息。所以很多開源算法僅僅能提供人的前臉重建效果，并不是整頭的重建。

發明內容

提供了一種模型訓練、人頭重建方法，裝置，設備以及存儲介質。

根據第一方面，提供了一種模型訓練方法，包括：獲取對預先制作的人頭模型進行展開得到的展開紋理圖像；獲取多張樣本二維人臉圖像；根據多張樣本二維人臉圖像以及展開紋理圖像，確定各樣本二維人臉圖像對應的整頭紋理圖像；將各樣本二維人臉圖像作為輸入，將與所輸入的樣本二維人臉圖像對應的整頭紋理圖像作為期望輸出，訓練得到整頭紋理圖像生成模型。

根據第二方面，提供了一種人頭重建方法，包括：獲取目標二維人臉圖像；獲取與目標二維人臉圖像對應的白模三維人頭模型；基于目標二維人臉圖像以及預先訓練的整頭紋理圖像生成模型，確定目標整頭紋理圖像，整頭紋理圖像生成模型通過如第一方面所描述的模型訓練方法訓練得到；根據目標整頭紋理圖像以及白模三維人頭模型，確定目標三維人頭模型。

根據第三方面，提供了一種模型訓練裝置，包括：第一獲取單元，被配置成獲取對預先制作的人頭模型進行展開得到的展開紋理圖像；第二獲取單元，被配置成獲取多張樣本二維人臉圖像；圖像處理單元，被配置成根據多張樣本二維人臉圖像以及展開紋理圖像，確定各樣本二維人臉圖像對應的整頭紋理圖像；模型訓練單元，被配置成將各樣本二維人臉圖像作為輸入，將與所輸入的樣本二維人臉圖像對應的整頭紋理圖像作為期望輸出，訓練得到整頭紋理圖像生成模型。

根據第四方面，提供了一種人頭重建裝置，包括：第三獲取單元，被配置成獲取目標二維人臉圖像；第四獲取單元，被配置成獲取與目標二維人臉圖像對應的白模三維人頭模型；圖像生成單元，被配置成基于目標二維人臉圖像以及預先訓練的整頭紋理圖像生成模型，確定目標整頭紋理圖像，整頭紋理圖像生成模型通過如第一方面所描述的模型訓練方法訓練得到；人頭重建單元，被配置成根據目標整頭紋理圖像以及白模三維人頭模型，確定目標三維人頭模型。

根據第五方面，提供了一種執行模型訓練方法的電子設備，包括：至少一個處理器；以及與上述至少一個處理器通信連接的存儲器；其中，存儲器存儲有可被至少一個處理器執行的指令，上述指令被至少一個處理器執行，以使至少一個處理器能夠執行如第一方面所描述的方法。

根據第六方面，提供了一種執行人頭重建方法的電子設備，包括：至少一個處理器；以及與上述至少一個處理器通信連接的存儲器；其中，存儲器存儲有可被至少一個處理器執行的指令，上述指令被至少一個處理器執行，以使至少一個處理器能夠執行如第二方面所描述的方法。

根據第七方面，提供了一種存儲有計算機指令的非瞬時計算機可讀存儲介質，上述計算機指令用于使計算機執行如第一方面所描述的方法或如第二方面所描述的方法。

根據第八方面，一種計算機程序產品，包括計算機程序，上述計算機程序在被處理器執行時實現如第一方面所描述的方法或如第二方面所描述的方法。