[發明專利]一種結合注意力機制和部分親和域場的人體姿態估計方法

在審

• 申請號: 202010614346.6
• 申請日: 2020-06-30
• 公開（公告）號: CN111860216A
• 公開（公告）日: 2020-10-30
• 發明（設計）人: 盧健;楊騰飛;周嫣然;羅毛欣;李哲
• 申請（專利權）人: 西安工程大學
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 西安弘理專利事務所 61214
• 代理人: 韓玙

1.一種結合注意力機制和部分親和域場的人體姿態估計方法，其特征在于，具體按照以下步驟實施：

步驟1，獲取人體姿態估計的公用數據集；

步驟2，將公用數據集中的待測圖像輸入到沙漏堆棧網絡，通過多語境注意力模型得到人體全局注意力圖；

步驟3，將所述人體全局注意力圖輸入到多階段雙分支網絡；

步驟4，采用損失函數引導所述多階段雙分支網絡對人體全局注意力圖進行預測并迭代，直至多階段雙分支網絡收斂，獲得人體局部注意力圖和部分親和域場；

步驟5，對所述人體局部注意力圖和部分親和域場進行聚類，得到待測圖像中人體姿態估計結果。

2.根據權利要求1所述的一種結合注意力機制和部分親和域場的人體姿態估計方法，其特征在于，步驟1中，所述公用數據集為MSCOCO數據集、MPII數據集或LSP數據集。

3.根據權利要求1所述的一種結合注意力機制和部分親和域場的人體姿態估計方法，其特征在于，步驟4中，所述多階段雙分支網絡包括第一網絡分支和第二網絡分支，所述多階段包括第一階段和第二階段；

所述第一階段是通過沙漏堆棧網絡將待測圖像中人體區域與背景分離開來，單獨提取出待測圖像中的人體區域；所述第二階段是分別通過第一網絡分支和第二網絡分支對所述人體區域進行預測；

所述第一網絡分支預測人體區域中的人體關節點，獲得人體局部注意力圖；所述第二網絡分支預測人體區域中的人體軀干，獲得部分親和域場。

4.根據權利要求1所述的一種結合注意力機制和部分親和域場的人體姿態估計方法，其特征在于，步驟4中，所述損失函數具體表述為：

式(1)中，為局部置信度的誤差情況；為部分親和域場的誤差情況。

5.根據權利要求4所述的一種結合注意力機制和部分親和域場的人體姿態估計方法，其特征在于，所述損失函數中，

式(2)中，為局部置信度真值；p為待測圖像中的像素；為在t階段預測的局部置信度值；j∈(1.,.,.,J)，J為局部注意力圖的總數；W(p)為像素p缺少注釋時的二進制掩碼；

式(3)中，為部分親和域真值；為在t階段預測的部分親和域；c∈(1，.,.,.,C)，C為親和域場總數。