[發明專利]頭盔識別方法、裝置、設備及存儲介質

有效

• 申請號: 202110516194.0
• 申請日: 2021-05-12
• 公開（公告）號: CN113159036B
• 公開（公告）日: 2022-07-12
• 發明（設計）人: 梁婷;肖瀟;孟祥昊;付馬;肖賀;丁鑫
• 申請（專利權）人: 蘇州科達科技股份有限公司
• 主分類號: G06V10/25
• 分類號: G06V10/25;G06V10/75;G06V10/774;G06V10/82;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 蘇州謹和知識產權代理事務所(特殊普通合伙) 32295
• 代理人: 葉棟

本申請涉及一種頭盔識別方法、裝置、設備及存儲介質，屬于計算機技術領域，該方法包括：獲取目標圖像；將目標圖像輸入預先訓練的頭盔識別網絡，得到識別結果；基于頭盔區域的檢測結果和頭肩區域的檢測結果進行頭盔識別；可以解決現有的頭盔識別效率較低的問題；頭盔識別網絡具有同時檢測頭盔區域和頭肩區域的功能，這樣，可以同時實現頭盔區域和頭肩區域的檢測，并結合兩個區域的檢測結果識別頭盔，既可以保證頭盔識別的準確性，又可以提高頭盔識別效率。

【技術領域】

本申請涉及一種頭盔識別方法、裝置、設備及存儲介質，屬于計算機技術領域。

【背景技術】

隨著對交通安全關注度的提升，如何對摩托車和電動車的騎行者是否佩戴頭盔進行識別是亟需解決的問題。

傳統的頭盔識別方式需要先識別騎行人員的頭肩區域，然后對該頭肩區域進行頭盔識別。

然而，上述方式需要先識別出頭肩區域，再對頭肩區域進一步識別才能得到頭盔識別結果，頭盔識別的效率較低。

【發明內容】

本申請提供了一種頭盔識別方法、裝置、設備及存儲介質，可以解決現有的頭盔識別效率較低的問題。本申請提供如下技術方案：

第一方面，提供一種頭盔識別方法，所述方法包括：

獲取目標圖像；

將所述目標圖像輸入預先訓練的頭盔識別網絡，得到識別結果；所述頭盔識別網絡具有同時檢測頭盔區域和頭肩區域的檢測功能，所述識別結果包括所述目標圖像中頭盔區域的檢測結果和頭肩區域的檢測結果；

基于所述頭盔區域的檢測結果和所述頭肩區域的檢測結果進行頭盔識別。

可選地，所述基于所述頭盔區域的檢測結果和所述頭肩區域的檢測結果進行頭盔識別，包括：

在所述頭盔區域的檢測結果指示存在所述頭盔區域、且所述頭肩區域的檢測結果指示存在所述頭肩區域的情況下，確定所述頭盔區域與所述頭肩區域之間的匹配關系，以確定所述頭盔區域與所述頭肩區域是否屬于同一騎行人員；

在匹配出所述頭盔區域與所述頭肩區域屬于同一騎行人員的情況下，確定所述頭肩區域對應的騎行人員佩戴頭盔。

可選地，所述頭盔識別網絡包括第一網絡分支和第二網絡分支；

所述第一網絡分支用于進行目標檢測，所述目標檢測包括頭盔的檢測和頭肩的檢測；

所述第二網絡分支用于進行屬性檢測，得到屬性信息；所述屬性信息用于輔助進行頭盔識別；

相應地，所述識別結果包括所述目標圖像的屬性信息。

可選地，所述基于所述頭盔區域的檢測結果和所述頭肩區域的檢測結果進行頭盔識別，包括：

按照所述屬性信息確定所述目標圖像中的車輛是否為預設類型，預設類型的車輛為需要騎行人員佩戴頭盔的車輛；其中，所述屬性信息包括車輛的類型屬性、載人數量屬性、頭盔的顏色屬性和/或頭盔的款式屬性；

在所述屬性信息確定所述目標圖像中的車輛為預設類型時，按照所述頭盔區域的檢測結果和所述頭肩區域的檢測結果確定所述頭盔區域與所述頭肩區域是否匹配，以確定所述頭盔區域與所述頭肩區域是否屬于同一騎行人員；

在所述頭盔區域與所述頭肩區域匹配時，確定所述目標圖像中所述頭肩區域對應的騎行人員佩戴頭盔。

可選地，所述頭盔識別網絡基于中心點的檢測網絡CenterNet建立，所述頭盔識別網絡從前至后依次包括主干網絡、分別與所述主干網絡相連的所述第一網絡分支和所述第二網絡分支；

所述第一網絡分支包括用于提取所述目標圖像的熱度圖；并基于所述熱度圖進行目標檢測；