[發明專利]一種人體數據集生成方法及裝置

在審

• 申請號: 202010414665.2
• 申請日: 2020-05-15
• 公開（公告）號: CN113689578A
• 公開（公告）日: 2021-11-23
• 發明（設計）人: 李順枝;許婭彤;李杰;毛慧
• 申請（專利權）人: 杭州海康威視數字技術股份有限公司
• 主分類號: G06T19/20
• 分類號: G06T19/20;G06T15/00;G06T17/00;G01B11/24;G01B11/16
• 代理公司: 北京柏杉松知識產權代理事務所(普通合伙) 11413
• 代理人: 馬敬;丁蕓

本申請實施例提供了一種人體數據集生成方法及裝置，獲取人體三維模型、背景圖像及指定形變參數，根據指定形變參數，對人體三維模型進行人體形變，得到形變后的人體三維模型，將形變后的人體三維模型渲染至背景圖像中，得到渲染圖像，并根據預設的任務類型，對渲染圖像進行人體標注得到標注信息，對應存儲渲染圖像及標注信息，得到人體數據集。僅需獲取少量的人體三維模型和背景圖像，通過設置不同的指定形變參數，可以得到不同的指定形變參數下形變后的人體三維模型，進而通過圖像渲染，能夠得到多張包括不同人體行為/姿態的渲染圖像，只需對應存儲渲染圖像和該渲染圖像中人體的標注信息，即可生成人體數據集，因此，能夠快速生成人體數據集。

技術領域

本申請涉及計算機視覺技術領域，特別是涉及一種人體數據集生成方法及裝置。

背景技術

隨著人工智能技術的不斷發展，在行人監控、行人跟蹤等領域中，對于人體檢測的需求不斷增加，為了實現對不同應用場景中的人體檢測功能，計算機可以通過人體檢測模型進行人體檢測，得到人體的標注信息，標注信息例如人體的行為/姿態類型、人體位置等。其中，人體檢測模型由計算機基于人體數據集，對初始檢測模型進行訓練得到；初始檢測模型例如支持向量機、隨機森林等，人體數據集包括多張圖像以及每張圖像中人體的標注信息。

相關技術中，訓練初始檢測模型所需的人體數據集由人工生成，具體處理過程為：針對多種需要進行人體檢測的應用場景，工作人員拍攝該應用場景中的人體，得到包含該人體的圖像。然后，人工標注人體的標注信息，例如，人工標注人體在該圖像中的位置、人體的行為類型/姿態類型，得到該應用場景的樣本圖像。由此，得到由每種應用場景的樣本圖像組成的人體數據集。

為了保證人體檢測模型檢測結果的準確率，需要基于優質人體數據集對初始檢測模型進行訓練，優質人體數據集是指包含的樣本圖像的數量大、且樣本圖像所包含人體的行為/姿態多樣的數據集。然而，通過人工方式生成人體數據集的效率低，無法快速生成優質人體數據集。

發明內容

本申請實施例的目的在于提供一種人體數據集生成方法及裝置，以實現快速生成人體數據集。具體技術方案如下：

第一方面，本申請實施例提供了一種人體數據集生成方法，該方法包括：

獲取人體三維模型、背景圖像及指定形變參數；

根據指定形變參數，對人體三維模型進行人體形變，得到形變后的人體三維模型；

將形變后的人體三維模型渲染至背景圖像中，得到渲染圖像，并根據預設的任務類型，對渲染圖像進行人體標注得到標注信息；

對應存儲渲染圖像及標注信息，得到人體數據集。

可選的，獲取人體三維模型的步驟，包括：

獲取多臺采集設備采集的多視角圖像和深度圖像，以及預先對多臺采集設備進行參數標定得到的各采集設備的內參和各采集設備之間的外參；

根據各采集設備的內參及各采集設備之間的外參，將多視角圖像中的各視角圖像和深度圖像對齊，并將對齊后的各視角圖像和深度圖像投影至世界坐標系，得到各視角三維點云；

將各視角三維點云進行對齊，并去除對齊后的各視角三維點云中共視的重復三維點云，得到融合后的多視角三維點云；

對多視角三維點云進行表面模型重建，得到網絡三維模型；

將多視角圖像映射至網絡三維模型，得到人體三維模型。

可選的，在根據指定形變參數，對人體三維模型進行人體形變，得到形變后的人體三維模型的步驟之前，該方法還包括：

對人體三維模型進行擬合，得到人體三維模型對應的參數化模型；