[發明專利]一種人體關鍵點檢測方法和系統

在審

• 申請號: 202210086523.7
• 申請日: 2022-01-25
• 公開（公告）號: CN114565937A
• 公開（公告）日: 2022-05-31
• 發明（設計）人: 王海;李東東
• 申請（專利權）人: 重慶特斯聯啟智科技有限公司
• 主分類號: G06V40/10
• 分類號: G06V40/10;G06V10/44;G06V10/82;G06V10/80;G06K9/62;G06T3/40;G06N3/04
• 代理公司: 北京辰權知識產權代理有限公司 11619
• 代理人: 李小朋

1.一種人體關鍵點檢測方法，其特征在于，所述方法包括：

對待檢測圖像進行縮放處理；

將縮放處理后的圖像輸入卷積神經網絡中，得到人體關鍵點的熱圖特征圖；所述卷積神經網絡包括第一數目個倒置殘差單元和第二數目個反卷積單元；

分別對所述熱圖特征圖中每個人體關鍵點通道取最大值，以定位熱圖特征圖中的人體關鍵點位置；

基于所述熱圖特征圖中的人體關鍵點位置以及所述熱圖特征圖和所述待檢測圖像的大小比例關系，確定所述待檢測圖像中的人體關鍵點位置。

2.根據權利要求1所述的方法，其特征在于，所述卷積神經網絡還包括至少一個融合單元；

所述第一數目個倒置殘差單元依次連接，形成倒置殘差分支；所述第二數目個反卷積單元依次連接，形成反卷積分支；所述倒置殘差分支與所述反卷積分支連接；

每個所述融合單元分別位于相鄰的兩個反卷積單元之間，每個融合單元分別與一個倒置殘差單元連接；

所述融合單元，用于對其輸入端連接的倒置殘差單元和反卷積單元輸出的特征圖進行通道融合，將得到的融合特征圖輸入所述融合單元的輸出端連接的反卷積單元。

3.如權利要求1所述的方法，其特征在于，所述倒置殘差單元具體用于：

通過第一卷積層增加輸入的特征圖像的通道數；

通過第二卷積層對增加通道數后的特征圖進行可分離卷積；

通過第三卷積層將第二卷積層輸出的特征圖的通道數降低到與所述輸入的特征圖像的通道數相同；

將所述第三卷積層輸出的特征圖與所述輸入的特征圖像相加，得到所述倒置殘差單元輸出的特征圖。

4.如權利要求1所述的方法，其特征在于，所述熱圖特征圖的通道數是設定的人體關鍵點的數目。

5.如權利要求1所述的方法，其特征在于，所述分別對所述熱圖特征圖中每個人體關鍵點通道取最大值，以定位熱圖特征圖中的人體關鍵點位置，包括：

分別對所述熱圖特征圖中每個人體關鍵點通道在寬和高的維度上取最大值，將最大值所在坐標位置確定為所述熱圖特征圖中的人體關鍵點的位置，以定位熱圖特征圖中的所有人體關鍵點位置。

6.一種人體關鍵點檢測系統，其特征在于，所述系統包括：

縮圖模塊，用于對待檢測圖像進行縮放處理；

卷積模塊，用于將縮放處理后的圖像輸入卷積神經網絡中，得到人體關鍵點的熱圖特征圖；所述卷積神經網絡包括第一數目個倒置殘差單元和第二數目個反卷積單元；

關鍵點第一定位模塊，用于分別對所述熱圖特征圖中每個人體關鍵點通道取最大值，以定位熱圖特征圖中的人體關鍵點位置；

關鍵點第二定位模塊，用于基于所述熱圖特征圖中的人體關鍵點位置以及所述熱圖特征圖和所述待檢測圖像的大小比例關系，確定所述待檢測圖像中的人體關鍵點位置。

7.如權利要求6所述的系統，其特征在于，所述卷積神經網絡還包括至少一個融合單元；

所述第一數目個倒置殘差單元依次連接，形成倒置殘差分支；所述第二數目個反卷積單元依次連接，形成反卷積分支；所述倒置殘差分支與所述反卷積分支連接；

每個所述融合單元分別位于相鄰的兩個反卷積單元之間，每個融合單元分別與一個倒置殘差單元連接；

所述融合單元，用于對其輸入端連接的倒置殘差單元和反卷積單元輸出的特征圖進行通道融合，將得到的融合特征圖輸入所述融合單元的輸出端連接的反卷積單元。

8.如權利要求6所述的系統，其特征在于，所述倒置殘差單元具體用于：

通過第一卷積層增加輸入的特征圖像的通道數；

通過第二卷積層對增加通道數后的特征圖進行可分離卷積；

通過第三卷積層將第二卷積層輸出的特征圖的通道數降低到與所述輸入的特征圖像的通道數相同；

將所述第三卷積層輸出的特征圖與所述輸入的特征圖像相加，得到所述倒置殘差單元輸出的特征圖。