[發明專利]人體關鍵點識別方法和裝置

有效

• 申請號: 202010327619.9
• 申請日: 2020-04-23
• 公開（公告）號: CN111523468B
• 公開（公告）日: 2023-08-08
• 發明（設計）人: 盧子鵬;王健;文石磊;孫昊;丁二銳;章宏武
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06V40/10
• 分類號: G06V40/10;G06V10/40;G06V10/774
• 代理公司: 北京英賽嘉華知識產權代理有限責任公司 11204
• 代理人: 王達佐;馬曉亞

1.一種人體關鍵點識別方法，包括：

獲取待識別人體關鍵點的目標圖像；

將所述目標圖像輸入至預先訓練的人體關鍵點檢測模型，得到所述目標圖像中每個人體關鍵點的熱度圖，所述預先訓練的人體關鍵點檢測模型基于標注有每個人體關鍵點的真值熱度圖的樣本圖像訓練得到，其中，所述每個人體關鍵點的真值熱度圖的徑向作用范圍基于尺度參數確定，所述尺度參數根據尺度參數所對應的人體關鍵點與其余人體關鍵點之間的坐標的距離值確定，包括：若所述尺度參數所對應的人體關鍵點與其余人體關鍵點之間的坐標的距離值中的最小值小于預設距離值，則尺度參數根據所述最小值確定，若所述尺度參數所對應的人體關鍵點與其余人體關鍵點之間的坐標的距離值中的最小值大于等于預設距離值，則尺度參數為預設固定值；

根據所述目標圖像中每個人體關鍵點的熱度圖，確定每個人體關鍵點的坐標。

2.根據權利要求1所述的方法，其中，所述標注有每個人體關鍵點的真值熱度圖的樣本圖像包括：標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像；以及所述標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像通過以下方式得到：

獲取原始圖像，所述原始圖像包括標注有每個人體關鍵點的坐標的人體圖像及人體檢測框；

在所述原始圖像中，截取以人體檢測框的中心為中心，以人體檢測框為邊界的第一預設尺寸的圖像，得到第一圖像；

將第一圖像中的人體檢測框外擴第二預設尺寸，得到第二圖像；

在所述第二圖像中隨機截取第一預設尺寸的圖像，得到標注有每個人體關鍵點的坐標的截斷人體的圖像；

根據所述標注有每個人體關鍵點的坐標的截斷人體的圖像，得到標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像。

3.根據權利要求2所述的方法，其中，所述標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像的人體關鍵點的數量大于等于預設數量值。

4.根據權利要求2所述的方法，其中，在所述在所述原始圖像中，截取以人體檢測框的中心為中心，以人體檢測框為邊界的第一預設尺寸的圖像，得到第一圖像之前，所述方法還包括：

對人體檢測框進行隨機概率和隨機角度的旋轉。

5.一種人體關鍵點識別裝置，包括：

獲取模塊，被配置成獲取待識別人體關鍵點的目標圖像；

訓練模塊，被配置成將所述目標圖像輸入至預先訓練的人體關鍵點檢測模型，得到所述目標圖像中每個人體關鍵點的熱度圖，所述預先訓練的人體關鍵點檢測模型基于標注有每個人體關鍵點的真值熱度圖的樣本圖像訓練得到，其中，所述每個人體關鍵點的真值熱度圖的徑向作用范圍基于尺度參數確定，所述尺度參數根據尺度參數所對應的人體關鍵點與其余人體關鍵點之間的坐標的距離值確定，包括：若所述尺度參數所對應的人體關鍵點與其余人體關鍵點之間的坐標的距離值中的最小值小于預設距離值，則尺度參數根據所述最小值確定，若所述尺度參數所對應的人體關鍵點與其余人體關鍵點之間的坐標的距離值中的最小值大于等于預設距離值，則尺度參數為預設固定值；

確定模塊，被配置成根據所述目標圖像中每個人體關鍵點的熱度圖，確定每個人體關鍵點的坐標。

6.根據權利要求5所述的裝置，其中，所述標注有每個人體關鍵點的真值熱度圖的樣本圖像包括：標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像；以及所述標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像通過以下方式得到：

獲取原始圖像，所述原始圖像包括標注有每個人體關鍵點的坐標的人體圖像及人體檢測框；

在所述原始圖像中，截取以人體檢測框的中心為中心，以人體檢測框為邊界的第一預設尺寸的圖像，得到第一圖像；

將第一圖像中的人體檢測框外擴第二預設尺寸，得到第二圖像；

在所述第二圖像中隨機截取第一預設尺寸的圖像，得到標注有每個人體關鍵點的坐標的截斷人體的圖像；

根據所述標注有每個人體關鍵點的坐標的截斷人體的圖像，得到標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像。

7.根據權利要求6所述的裝置，其中，所述標注有每個人體關鍵點的真值熱度圖的截斷人體的圖像的人體關鍵點的數量大于等于預設數量值。