[發(fā)明專利]關鍵點定位方法、裝置、設備以及存儲介質

有效

• 申請?zhí)枺?/td>: 202010538339.2
• 申請日: 2020-06-12
• 公開（公告）號: CN111695519B
• 公開（公告）日: 2023-08-08
• 發(fā)明（設計）人: 王健;盧子鵬;孫昊;章宏武;文石磊;丁二銳
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06V10/25
• 分類號: G06V10/25;G06V40/20;G06V10/52;G06V10/82;G06N3/0464
• 代理公司: 北京英賽嘉華知識產權代理有限責任公司 11204
• 代理人: 王達佐;馬曉亞

1.一種關鍵點定位方法，包括：

提取待定位圖像的第一特征圖和第二特征圖，其中，所述第一特征圖與所述第二特征圖是不同的特征圖；

基于所述第一特征圖，確定所述待定位圖像中的關鍵點的初始位置，其中，利用第一特征圖中的第一尺度的全局語義特征，確定所述關鍵點的初始位置；

基于所述第二特征圖，確定所述關鍵點的偏移量，其中，利用所述第二特征圖中的第二尺度的局部語義特征，確定所述關鍵點的偏移量，所述第一尺度大于所述第二尺度；

將所述關鍵點的初始位置與所述關鍵點的偏移量相加，得到所述關鍵點的最終位置；

其中，所述提取待定位圖像的第一特征圖和第二特征圖，包括：

將所述待定位特征圖輸入至主網絡，輸出所述待定位圖像的初始特征圖；

將所述初始特征圖分別輸入至第一子網絡和第二子網絡，輸出所述第一特征圖和所述第二特征圖，其中，所述第一子網絡與所述第二子網絡是所述主網絡的兩個不同的分支。

2.根據權利要求1所述的方法，其中，所述基于所述第一特征圖，確定所述待定位圖像中的關鍵點的初始位置，包括：

基于所述第一特征圖，生成所述待定位圖像中的關鍵點的熱力圖；

基于所述熱力圖上的點的熱力值，確定所述關鍵點的初始位置。

3.根據權利要求2所述的方法，其中，所述基于所述第一特征圖，生成所述待定位圖像中的關鍵點的熱力圖，包括：

將所述第一特征圖經過1乘1卷積，得到所述熱力圖，其中，所述熱力圖的通道與所述關鍵點一一對應。

4.根據權利要求1所述的方法，其中，所述基于所述第二特征圖，確定所述關鍵點的偏移量，包括：

基于所述關鍵點的初始位置，在所述第二特征圖的對應位置上提取特征；

利用所述特征進行偏移量回歸，得到所述關鍵點的偏移量。

5.一種關鍵點定位裝置，包括：

特征圖提取模塊，被配置成提取待定位圖像的第一特征圖和第二特征圖，其中，所述第一特征圖與所述第二特征圖是不同的特征圖；

初始位置確定模塊，被配置成基于所述第一特征圖，確定所述待定位圖像中的關鍵點的初始位置，其中，利用第一特征圖中的較大尺度的全局語義特征，確定所述關鍵點的初始位置；

偏移量確定模塊，被配置成基于所述第二特征圖，確定所述關鍵點的偏移量，其中，利用所述第二特征圖中的較小尺度的局部語義特征，確定所述關鍵點的偏移量；

最終位置生成模塊，被配置成將所述關鍵點的初始位置與所述關鍵點的偏移量相加，得到所述關鍵點的最終位置；

其中，所述特征圖提取模塊進一步被配置成：

將所述待定位特征圖輸入至主網絡，輸出所述待定位圖像的初始特征圖；

將所述初始特征圖分別輸入至第一子網絡和第二子網絡，輸出所述第一特征圖和所述第二特征圖，其中，所述第一子網絡與所述第二子網絡是所述主網絡的兩個不同的分支。

6.根據權利要求5所述的裝置，其中，所述初始位置確定模塊包括：

熱力圖生成子模塊，被配置成基于所述第一特征圖，生成所述待定位圖像中的關鍵點的熱力圖；

初始位置確定子模塊，被配置成基于所述熱力圖上的點的熱力值，確定所述關鍵點的初始位置。

7.根據權利要求6所述的裝置，其中，所述熱力圖生成子模塊進一步被配置成：

將所述第一特征圖經過1乘1卷積，得到所述熱力圖，其中，所述熱力圖的通道與所述關鍵點一一對應。

8.根據權利要求5所述的裝置，其中，所述偏移量確定模塊進一步被配置成：

基于所述關鍵點的初始位置，在所述第二特征圖的對應位置上提取特征；

利用所述特征進行偏移量回歸，得到所述關鍵點的偏移量。

9.一種電子設備，包括：

一個或多個處理器；

存儲裝置，其上存儲有一個或多個程序，

當所述一個或多個程序被所述一個或多個處理器執(zhí)行，使得所述一個或多個處理器實現(xiàn)如權利要求1-4中任一所述的方法。

10.一種計算機可讀介質，其上存儲有計算機程序，其中，所述計算機程序被處理器執(zhí)行時實現(xiàn)如權利要求1-4中任一所述的方法。