[發明專利]訓練方法、裝置、動作識別方法、裝置及電子設備

有效

• 申請號: 202110227119.2
• 申請日: 2021-03-01
• 公開（公告）號: CN112926462B
• 公開（公告）日: 2023-02-07
• 發明（設計）人: 張發恩;秦樹鑫
• 申請（專利權）人: 創新奇智（西安）科技有限公司
• 主分類號: G06V40/10
• 分類號: G06V40/10;G06V10/764;G06V10/44;G06V10/82;G06V10/766;G06N3/08
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 張萌

1.一種訓練方法，其特征在于，用于訓練輪廓特征提取網絡，所述方法包括：

獲取包括多個訓練樣本的訓練集，所述訓練樣本的訓練標簽包括：對象的中心點、所述對象的輪廓線、以所述中心點為起點且向所述輪廓線發散出N條射線后與所述輪廓線相交形成的N個輪廓關鍵點，相鄰兩條射線之間的角度差為A，且N＝360/A；

將所述訓練樣本輸入到基礎網絡模型進行訓練，直至所述基礎網絡模型收斂，得到所述輪廓特征提取網絡；

其中，所述輪廓特征提取網絡的輸出為與其輸入特征圖中的對象對應的輪廓特征圖，所述輪廓特征圖包括：與所述輸入特征圖中的對象的N個輪廓關鍵點一一對應的N個關鍵點通道、與所述輸入特征圖中的對象的中心點對應的中心點通道，以及與所述N個關鍵點通道一一對應的，用于表征該關鍵點通道所指向的輪廓關鍵點與所述輸入特征圖中的對象的中心點之間的距離回歸通道。

2.根據權利要求1所述的方法，其特征在于，當在所述輸入特征圖中包括多個對象時，所述輪廓特征提取網絡的輸出為與每個對象對應的輪廓特征圖。

3.一種動作識別方法，其特征在于，所述方法包括：

對獲取到的待識別圖像進行特征提取，得到對應的特征圖；

將所述特征圖輸入到通過權利要求1或2所述的方法訓練得到的輪廓特征提取網絡，輸出得到輪廓特征圖；

對所述輪廓特征圖進行解碼，得到輪廓特征向量；

將所述輪廓特征向量輸入到預先訓練好的分類模型進行分類，得到用于表征動作類別的分類結果。

4.根據權利要求3所述的方法，其特征在于，所述輪廓特征圖包括多個通道，每個通道對應一個熱力圖，所述對所述輪廓特征圖進行解碼，得到輪廓特征向量，包括：

從所述多個通道中，確定出用于表征所述特征圖中的對象的中心點的中心點通道、用于表征所述特征圖中的對象的輪廓關鍵點的關鍵點通道，以及與每個所述關鍵點通道一一對應的距離回歸通道；

根據與所述中心點通道對應的熱力圖以及與每個所述關鍵點通道對應的熱力圖，確定出所述中心點的坐標以及每個所述關鍵點通道所指向的輪廓關鍵點的坐標；

針對每個所述關鍵點通道，根據該關鍵點通道所指向的輪廓關鍵點的坐標以及所述中心點的坐標，從與該關鍵點通道對應的距離回歸通道所對應的熱力圖中，確定出該關鍵點通道所指向的輪廓關鍵點與所述中心點之間的距離；

根據各個輪廓關鍵點與所述中心點之間的距離，形成所述輪廓特征向量。

5.根據權利要求4所述的方法，其特征在于，所述根據與所述中心點通道對應的熱力圖以及與每個所述關鍵點通道對應的熱力圖，確定出所述中心點的坐標以及每個所述關鍵點通道所指向的輪廓關鍵點的坐標，包括：

將與所述中心點通道對應的熱力圖中，極大值對應的位置的坐標確定為所述中心點的坐標；

將與所述該關鍵點通道對應的熱力圖中，極大值對應的位置的坐標確定為所述該關鍵點通道所指向的輪廓關鍵點的坐標。

6.根據權利要求4所述的方法，其特征在于，所述根據該關鍵點通道所指向的輪廓關鍵點的坐標以及所述中心點的坐標，從與該關鍵點通道對應的距離回歸通道所對應的熱力圖中，確定出該關鍵點通道所指向的輪廓關鍵點與所述中心點之間的距離，包括：

獲取與該關鍵點通道對應的距離回歸通道對應的熱力圖中，與該關鍵點通道所指向的輪廓關鍵點的坐標處的第一取值以及所述中心點的坐標處的第二取值；

在確定所述第一取值大于所述第二取值時，確定所述第一取值為所述該關鍵點通道所指向的輪廓關鍵點與所述中心點之間的距離。

7.根據權利要求4所述的方法，其特征在于，預先為每個關鍵點通道設置對應的角度信息，所述根據各個輪廓關鍵點與所述中心點之間的距離，形成所述輪廓特征向量，包括：

根據各個關鍵點通道所對應的角度信息大小關系，對各個輪廓關鍵點與所述中心點之間的距離進行排序，形成所述輪廓特征向量。

8.根據權利要求3-7任一項所述的方法，其特征在于，所述待識別圖像為單幀圖像，或者所述待識別圖像為包括多幀圖像的圖像序列。