[發(fā)明專利]安全帽佩戴檢測模型處理方法、裝置、設(shè)備和存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202010581016.1
• 申請日: 2020-06-23
• 公開（公告）號: CN111815577A
• 公開（公告）日: 2020-10-23
• 發(fā)明（設(shè)計）人: 賴振宇;楊豐閣;高德民;徐紅星
• 申請（專利權(quán)）人: 深圳供電局有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T5/00;G06K9/62;G06K9/32;G06K9/00;G08B21/02
• 代理公司: 廣州華進聯(lián)合專利商標(biāo)代理有限公司 44224
• 代理人: 李文淵

本申請涉及一種安全帽佩戴檢測模型處理方法、裝置、計算機設(shè)備和存儲介質(zhì)。方法包括：獲取樣本圖像；樣本圖像包括人體頭部佩戴安全帽的圖像和人體頭部未佩戴安全帽的圖像；根據(jù)樣本圖像中人體頭部是否佩戴安全帽，對樣本圖像進行類別標(biāo)記，得到訓(xùn)練數(shù)據(jù)集和對應(yīng)的檢測類別數(shù)；獲取對目標(biāo)物體檢測數(shù)據(jù)集訓(xùn)練所得的預(yù)訓(xùn)練模型；預(yù)訓(xùn)練模型包括特征金字塔網(wǎng)絡(luò)和預(yù)測網(wǎng)絡(luò)；目標(biāo)物體檢測數(shù)據(jù)集包括的物體圖像中，物體的尺寸占比小于預(yù)設(shè)占比；將預(yù)訓(xùn)練模型的預(yù)測網(wǎng)絡(luò)的輸出類別數(shù)調(diào)整為檢測類別數(shù)，得到調(diào)整后預(yù)訓(xùn)練模型；將訓(xùn)練數(shù)據(jù)集輸入調(diào)整后預(yù)訓(xùn)練模型進行訓(xùn)練，得到安全帽佩戴檢測模型。采用本方法能夠提高對安全帽佩戴情況的監(jiān)控效率。

技術(shù)領(lǐng)域

本申請涉及計算機技術(shù)領(lǐng)域，特別是涉及一種安全帽佩戴檢測模型處理方法、裝置、設(shè)備和存儲介質(zhì)。

背景技術(shù)

安全帽對人體頭部受墜落物及其他特定因素引起的傷害具有較好的防護作用，為了避免安全事故的發(fā)生，在施工現(xiàn)場中要求工作人員佩戴安全帽是安全生產(chǎn)的必要措施。為了避免工作人員未佩戴安全帽進入施工現(xiàn)場，通常需要對施工現(xiàn)場的工作人員的安全帽佩戴情況進行監(jiān)控，傳統(tǒng)的監(jiān)控方式是通過攝像頭采集施工現(xiàn)場的圖像數(shù)據(jù)，根據(jù)該圖像數(shù)據(jù)確定出未佩戴安全帽的工作人員。

然而當(dāng)所采集的圖像數(shù)據(jù)的分辨率不高或圖像數(shù)據(jù)中的人體尺寸較小時，傳統(tǒng)的監(jiān)控方式無法對安全帽佩戴情況進行高效、準(zhǔn)確地判斷，從而導(dǎo)致對施工現(xiàn)場安全帽佩戴情況的監(jiān)控效率較低。

發(fā)明內(nèi)容

基于此，有必要針對施工現(xiàn)場安全帽佩戴情況的監(jiān)控效率低的技術(shù)問題，提供一種安全帽佩戴檢測模型處理方法、裝置、設(shè)備和存儲介質(zhì)。

一種安全帽佩戴檢測模型處理方法，所述方法包括：

獲取樣本圖像；所述樣本圖像包括人體頭部佩戴安全帽的圖像和人體頭部未佩戴安全帽的圖像；

根據(jù)所述樣本圖像中人體頭部是否佩戴安全帽，對所述樣本圖像進行類別標(biāo)記，得到訓(xùn)練數(shù)據(jù)集和對應(yīng)的檢測類別數(shù)；

獲取對目標(biāo)物體檢測數(shù)據(jù)集訓(xùn)練所得的預(yù)訓(xùn)練模型；所述預(yù)訓(xùn)練模型包括特征金字塔網(wǎng)絡(luò)和預(yù)測網(wǎng)絡(luò)；所述目標(biāo)物體檢測數(shù)據(jù)集包括的物體圖像中，物體的尺寸占比小于預(yù)設(shè)占比；

將所述預(yù)訓(xùn)練模型的預(yù)測網(wǎng)絡(luò)的輸出類別數(shù)調(diào)整為所述檢測類別數(shù)，得到調(diào)整后預(yù)訓(xùn)練模型；

將所述訓(xùn)練數(shù)據(jù)集輸入所述調(diào)整后預(yù)訓(xùn)練模型進行訓(xùn)練，得到安全帽佩戴檢測模型。

在一個實施例中，所述獲取樣本圖像，包括：

獲取人體頭部佩戴安全帽的原始圖像、人體頭部未佩戴安全帽的原始圖像；

對所述原始圖像進行光度變形和/或幾何變形的數(shù)據(jù)增強處理，得到預(yù)設(shè)數(shù)量的樣本圖像。

在一個實施例中，所述將所述訓(xùn)練數(shù)據(jù)集輸入所述調(diào)整后預(yù)訓(xùn)練模型進行訓(xùn)練，得到安全帽佩戴檢測模型，包括：

對所述訓(xùn)練數(shù)據(jù)集所包含的樣本圖像進行縮放處理，得到處理后訓(xùn)練數(shù)據(jù)集，所述處理后訓(xùn)練數(shù)據(jù)集所包含的樣本圖像的分辨率為預(yù)設(shè)分辨率；

將所述處理后訓(xùn)練數(shù)據(jù)集輸入所述調(diào)整后預(yù)訓(xùn)練模型進行訓(xùn)練，得到安全帽佩戴檢測模型。

在一個實施例中，訓(xùn)練所述安全帽佩戴檢測模型時，所使用的損失函數(shù)為：

L_focal＝-2α(1-y₁)^β(1-y^*)ln(y₁)-2(1-α)y^*ln(y₂)