[發明專利]圖像的標注方法、裝置及電子設備

在審

• 申請號: 202211630218.6
• 申請日: 2022-12-19
• 公開（公告）號: CN116129177A
• 公開（公告）日: 2023-05-16
• 發明（設計）人: 李磊;李源
• 申請（專利權）人: 中科智云科技有限公司;成都點澤智能科技有限公司
• 主分類號: G06V10/764
• 分類號: G06V10/764;G06V10/74;G06V10/42
• 代理公司: 成都極刻智慧知識產權代理事務所(普通合伙) 51310
• 代理人: 張紅平

本申請實施例提供的圖像的標注方法、裝置及電子設備，涉及圖像處理技術領域。首先，基于支撐圖像的特征圖與查詢圖像的特征圖，得到與查詢圖像的特征圖同形狀的第一內積變換結果；接著，基于第一內積變換結果得到同一類別目標物體的特征映射熱力圖；再接著，從特征映射熱力圖中確定出候選框，并采用候選框在查詢圖像中截取目標子圖；然后，基于目標子圖的特征向量和支撐圖像的特征向量，確定查詢圖像中目標物體和所述支撐圖像中目標物體的特征相似度；最后，根據特征相似度輸出查詢圖像中目標物體的邊框信息和類別。上述方案可以通過已標注的支撐圖像對大量的查詢圖像進行自動標注，相對于現有技術，標注的效率和精度都更高。

技術領域

本申請涉及圖像處理技術領域，具體而言，涉及一種圖像的標注方法、裝置及電子設備。

背景技術

某些特定場所對進入的目標物體是有明確的限制的，比如，高鐵站是禁止大型寵物進入的，目前針對此類目標物體的管控多半通過人工實現的，比如通過人工對進站區圖像進行標注的方式對目標物體進行標注以辨別該目標物體是否具有進入的權限，具體地，需要標注目標物體在進站區圖像中的位置區域和對應的類別，然而受標注人的專注程度的影響，標注人會隨時間而疲勞最終導致標注效率下降且錯誤增多。

發明內容

為了解決上述技術問題，本申請提供一種圖像的標注方法、裝置及電子設備。

第一方面，本申請實施例提供一種圖像的標注方法，所述方法包括：

對支撐集中的支撐圖像進行圖像特征提取，得到所述支撐圖像的特征向量和所述支撐圖像的特征圖，其中，所述支撐圖像包括標注信息，所述標注信息包括所述支撐圖像中目標物體的類別和用于將所述目標物體從所述支撐圖像中圈定出的邊框信息；

對查詢集中的查詢圖像進行圖像特征提取，得到所述查詢圖像的特征圖；

基于所述支撐圖像的特征圖與所述查詢圖像的特征圖，得到與所述查詢圖像的特征圖同形狀的第一內積變換結果；

基于所述第一內積變換結果得到同一類別目標物體的特征映射熱力圖；

從所述特征映射熱力圖中確定出候選框，并采用所述候選框在所述查詢圖像中截取目標子圖；

提取所述目標子圖的特征向量，基于所述目標子圖的特征向量和所述支撐圖像的特征向量，確定所述查詢圖像中目標物體和所述支撐圖像中目標物體的特征相似度；

根據所述查詢圖像中目標物體和所述支撐圖像中目標物體的特征相似度，輸出所述查詢圖像中目標物體的邊框信息和類別。

在一種可能的實現方式中，在從所述特征映射熱力圖中確定出候選框，并采用所述候選框在所述查詢圖像中截取目標子圖的步驟之后，所述方法還包括：

基于所述支撐圖像的特征圖與所述目標子圖的特征圖，得到與所述目標子圖的特征圖同形狀的第二內積變換結果；

基于所述第二內積變換結果更新同一類別的特征映射熱力圖；

基于更新后的特征映射熱力圖對所述候選框進行更新，由更新后的候選框重新在所述查詢圖像中截取目標子圖。

在一種可能的實現方式中，所述對支撐集中的支撐圖像進行圖像特征提取，得到所述支撐圖像的特征向量和所述支撐圖像的特征圖的步驟，包括：

將所述支撐集中的支撐圖像輸入圖像特征提取模型中，輸出所述支撐圖像的特征向量和所述支撐圖像的特征圖，并將所述支撐圖像的特征圖中所述邊框信息確定區域之外的區域設置為0；

所述對查詢集中的查詢圖像進行圖像特征提取，得到所述查詢圖像的特征圖的步驟，包括：

將所述查詢集中的查詢圖像輸入所述圖像特征提取模型中，輸出所述查詢圖像的特征圖。