[發(fā)明專利]頭盔識(shí)別方法、裝置、設(shè)備及存儲(chǔ)介質(zhì)

有效

• 申請(qǐng)?zhí)枺?/td>: 202110516194.0
• 申請(qǐng)日: 2021-05-12
• 公開(kāi)（公告）號(hào): CN113159036B
• 公開(kāi)（公告）日: 2022-07-12
• 發(fā)明（設(shè)計(jì)）人: 梁婷;肖瀟;孟祥昊;付馬;肖賀;丁鑫
• 申請(qǐng)（專利權(quán)）人: 蘇州科達(dá)科技股份有限公司
• 主分類號(hào): G06V10/25
• 分類號(hào): G06V10/25;G06V10/75;G06V10/774;G06V10/82;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 蘇州謹(jǐn)和知識(shí)產(chǎn)權(quán)代理事務(wù)所(特殊普通合伙) 32295
• 代理人: 葉棟

1.一種頭盔識(shí)別方法，其特征在于，所述方法包括：

獲取目標(biāo)圖像；

將所述目標(biāo)圖像輸入預(yù)先訓(xùn)練的頭盔識(shí)別網(wǎng)絡(luò)，得到識(shí)別結(jié)果；所述頭盔識(shí)別網(wǎng)絡(luò)具有同時(shí)檢測(cè)頭盔區(qū)域和頭肩區(qū)域的功能，所述識(shí)別結(jié)果包括所述目標(biāo)圖像中頭盔區(qū)域的檢測(cè)結(jié)果和頭肩區(qū)域的檢測(cè)結(jié)果；

基于所述頭盔區(qū)域的檢測(cè)結(jié)果和所述頭肩區(qū)域的檢測(cè)結(jié)果進(jìn)行頭盔識(shí)別；

所述頭盔識(shí)別網(wǎng)絡(luò)包括第一網(wǎng)絡(luò)分支和第二網(wǎng)絡(luò)分支；

所述第一網(wǎng)絡(luò)分支用于進(jìn)行目標(biāo)檢測(cè)，所述目標(biāo)檢測(cè)包括頭盔的檢測(cè)和頭肩的檢測(cè)；

所述第二網(wǎng)絡(luò)分支用于進(jìn)行屬性檢測(cè)，得到屬性信息；所述屬性信息用于輔助進(jìn)行頭盔識(shí)別；

相應(yīng)地，所述識(shí)別結(jié)果包括所述目標(biāo)圖像的屬性信息。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述基于所述頭盔區(qū)域的檢測(cè)結(jié)果和所述頭肩區(qū)域的檢測(cè)結(jié)果進(jìn)行頭盔識(shí)別，包括：

在所述頭盔區(qū)域的檢測(cè)結(jié)果指示存在所述頭盔區(qū)域、且所述頭肩區(qū)域的檢測(cè)結(jié)果指示存在所述頭肩區(qū)域的情況下，確定所述頭盔區(qū)域與所述頭肩區(qū)域之間的匹配關(guān)系，以確定所述頭盔區(qū)域與所述頭肩區(qū)域是否屬于同一騎行人員；

在匹配出所述頭盔區(qū)域與所述頭肩區(qū)域?qū)儆谕或T行人員的情況下，確定所述頭肩區(qū)域?qū)?yīng)的騎行人員佩戴頭盔。

3.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述基于所述頭盔區(qū)域的檢測(cè)結(jié)果和所述頭肩區(qū)域的檢測(cè)結(jié)果進(jìn)行頭盔識(shí)別，包括：

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

在所述屬性信息確定所述目標(biāo)圖像中的車(chē)輛為所述預(yù)設(shè)類型時(shí)，按照所述頭盔區(qū)域的檢測(cè)結(jié)果和所述頭肩區(qū)域的檢測(cè)結(jié)果確定所述頭盔區(qū)域與所述頭肩區(qū)域是否匹配，以確定所述頭盔區(qū)域與所述頭肩區(qū)域是否屬于同一騎行人員；

在所述頭盔區(qū)域與所述頭肩區(qū)域匹配時(shí)，確定所述目標(biāo)圖像中所述頭肩區(qū)域?qū)?yīng)的騎行人員佩戴頭盔。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述頭盔識(shí)別網(wǎng)絡(luò)基于中心點(diǎn)的檢測(cè)網(wǎng)絡(luò)CenterNet建立，所述頭盔識(shí)別網(wǎng)絡(luò)從前至后依次包括主干網(wǎng)絡(luò)、分別與所述主干網(wǎng)絡(luò)相連的所述第一網(wǎng)絡(luò)分支和所述第二網(wǎng)絡(luò)分支；

所述第一網(wǎng)絡(luò)分支包括用于提取所述目標(biāo)圖像的熱度圖；并基于所述熱度圖進(jìn)行目標(biāo)檢測(cè)；

所述第二網(wǎng)絡(luò)分支包括深度可分離卷積網(wǎng)絡(luò)，所述深度可分離卷積網(wǎng)絡(luò)的網(wǎng)絡(luò)層數(shù)小于第一閾值和/或參數(shù)數(shù)量小于第二閾值，所述第一閾值小于或等于現(xiàn)有的卷積網(wǎng)絡(luò)的最小層數(shù)；所述第二閾值小于或等于現(xiàn)有的卷積網(wǎng)絡(luò)的最小參數(shù)數(shù)量。

5.根據(jù)權(quán)利要求4所述的方法，其特征在于，所述主干網(wǎng)絡(luò)分別與所述第一網(wǎng)絡(luò)分支和所述第二網(wǎng)絡(luò)分支進(jìn)行局部殘差連接。

6.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述第二網(wǎng)絡(luò)分支基于改進(jìn)的focal loss函數(shù)訓(xùn)練得到，所述改進(jìn)的focal loss函數(shù)通過(guò)下式表示：

FL(p_t)＝-α_t(1-p_t)^γlog(p_t)

其中，pt表示網(wǎng)絡(luò)預(yù)測(cè)結(jié)果，F(xiàn)L(pt)表示所述改進(jìn)的focal loss函數(shù)的函數(shù)值，γ為預(yù)設(shè)常數(shù)，α_t為平衡因子，并通過(guò)下式表示：

其中，w是訓(xùn)練數(shù)據(jù)中具有屬性標(biāo)簽的樣本所占樣本總數(shù)的權(quán)重值。