[發(fā)明專利]安全帽檢測方法、裝置、電子設(shè)備及計算機可讀存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202210964056.3
• 申請日: 2022-08-11
• 公開（公告）號: CN115496708A
• 公開（公告）日: 2022-12-20
• 發(fā)明（設(shè)計）人: 李建強;趙琳娜;邸遠(yuǎn)航;程文秀
• 申請（專利權(quán)）人: 北京工業(yè)大學(xué)
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06V10/25;G06V10/56;G06V10/774;G06V10/82;G06V40/10
• 代理公司: 北京路浩知識產(chǎn)權(quán)代理有限公司 11002
• 代理人: 肖艷

本發(fā)明提供一種安全帽檢測方法、裝置、電子設(shè)備及計算機可讀存儲介質(zhì)，該方法包括：根據(jù)獲取的待處理圖像，基于人體姿態(tài)估計網(wǎng)絡(luò)，確定節(jié)點圖像；根據(jù)所述節(jié)點圖像中的人體關(guān)鍵節(jié)點，確定感興趣區(qū)域圖像；提取所述感興趣區(qū)域圖像中安全帽對應(yīng)的目標(biāo)顏色像素；根據(jù)所述目標(biāo)顏色像素，確定所述安全帽的佩戴情況。該方法用以解決現(xiàn)有技術(shù)中現(xiàn)有的安全帽檢測方法都具有一定的局限性，從而易導(dǎo)致電子設(shè)備無法準(zhǔn)確確定工作人員是否佩戴安全帽的缺陷，實現(xiàn)電子設(shè)備基于人體姿態(tài)估計網(wǎng)絡(luò)，可準(zhǔn)確確定人體關(guān)鍵節(jié)點，從而從待處理圖像中能夠準(zhǔn)確確定安全帽的佩戴情況。

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

本發(fā)明涉及計算機視覺技術(shù)領(lǐng)域，尤其涉及一種安全帽檢測方法、裝置、電子設(shè)備及計算機可讀存儲介質(zhì)。

背景技術(shù)

在工業(yè)生產(chǎn)過程中，工廠安全一直是生產(chǎn)建設(shè)工作的重中之重，尤其對于需要佩戴安全帽的環(huán)境，工作人員合理佩戴安全帽，可減輕和防止該工作人員在生產(chǎn)作業(yè)中遭受墜落物體對頭部的傷害，從而有效保障該工作人員的人身安全。

在現(xiàn)有的安全帽檢測方法中，電子設(shè)備往往采用傳統(tǒng)的深度學(xué)習(xí)網(wǎng)絡(luò)或利用人體姿態(tài)估計特征，確定工作人員是否佩戴安全帽。然而，電子設(shè)備采用該傳統(tǒng)的深度學(xué)習(xí)網(wǎng)絡(luò)可能無法獲取準(zhǔn)確的檢測區(qū)域，或該電子設(shè)備無法獲取準(zhǔn)確的人體姿態(tài)估計特征，也就是說，現(xiàn)有的安全帽檢測方法都具有一定的局限性，從而易導(dǎo)致該電子設(shè)備無法準(zhǔn)確確定工作人員是否佩戴安全帽。

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

本發(fā)明提供一種安全帽檢測方法、裝置、電子設(shè)備及計算機可讀存儲介質(zhì)，用以解決現(xiàn)有技術(shù)中現(xiàn)有的安全帽檢測方法都具有一定的局限性，從而易導(dǎo)致電子設(shè)備無法準(zhǔn)確確定工作人員是否佩戴安全帽的缺陷，實現(xiàn)電子設(shè)備基于人體姿態(tài)估計網(wǎng)絡(luò)，可準(zhǔn)確確定人體關(guān)鍵節(jié)點，從而從待處理圖像中能夠準(zhǔn)確確定安全帽的佩戴情況。

本發(fā)明提供一種安全帽檢測方法，包括：

根據(jù)獲取的待處理圖像，基于人體姿態(tài)估計網(wǎng)絡(luò)，確定節(jié)點圖像；

根據(jù)該節(jié)點圖像中的人體關(guān)鍵節(jié)點，確定感興趣區(qū)域圖像；

提取該感興趣區(qū)域圖像中安全帽對應(yīng)的目標(biāo)顏色像素；

根據(jù)該目標(biāo)顏色像素，確定該安全帽的佩戴情況。

根據(jù)本發(fā)明提供的一種安全帽檢測方法，該根據(jù)獲取的待處理圖像，基于人體姿態(tài)估計網(wǎng)絡(luò)，確定節(jié)點圖像，包括：獲取待處理圖像及該待處理圖像對應(yīng)的當(dāng)前亮度；在該當(dāng)前亮度小于預(yù)設(shè)亮度閾值的情況下，利用暗光圖像增強網(wǎng)絡(luò)增強該待處理圖像，得到增強圖像；根據(jù)該增強圖像，基于人體姿態(tài)估計網(wǎng)絡(luò)，確定節(jié)點圖像。

根據(jù)本發(fā)明提供的一種安全帽檢測方法，該根據(jù)獲取的待處理圖像，基于人體姿態(tài)估計網(wǎng)絡(luò)，確定節(jié)點圖像，包括：根據(jù)獲取的待處理圖像，基于人體姿態(tài)估計網(wǎng)絡(luò)中的目標(biāo)檢測網(wǎng)絡(luò)，確定包括待測人體的人體圖像；對該人體圖像中該待測人體對應(yīng)的關(guān)鍵點進行檢測和標(biāo)注，得到包括人體關(guān)鍵節(jié)點的節(jié)點圖像。

根據(jù)本發(fā)明提供的一種安全帽檢測方法，該根據(jù)該節(jié)點圖像中的人體關(guān)鍵節(jié)點，確定感興趣區(qū)域圖像，包括：從該節(jié)點圖像中的人體關(guān)鍵節(jié)點中，確定目標(biāo)節(jié)點；根據(jù)該目標(biāo)節(jié)點，確定感興趣區(qū)域圖像；其中，該目標(biāo)節(jié)點包括左肩膀節(jié)點、左臀節(jié)點、左耳節(jié)點及右耳節(jié)點；或，該目標(biāo)節(jié)點包括右肩膀節(jié)點、右臀節(jié)點、該左耳節(jié)點及該右耳節(jié)點；或，該目標(biāo)節(jié)點包括左肩膀節(jié)點、右肩膀節(jié)點、左膝蓋節(jié)點及左腳踝節(jié)點；或，該目標(biāo)節(jié)點包括該左肩膀節(jié)點、該右肩膀節(jié)點、右膝蓋節(jié)點及右腳踝節(jié)點。

根據(jù)本發(fā)明提供的一種安全帽檢測方法，該提取該感興趣區(qū)域圖像中安全帽對應(yīng)的目標(biāo)顏色像素，包括：將該感興趣區(qū)域圖像對應(yīng)的 RGB空間進行色彩空間轉(zhuǎn)換，得到HSV圖像；提取該HSV圖像中安全帽對應(yīng)的目標(biāo)顏色像素。

根據(jù)本發(fā)明提供的一種安全帽檢測方法，該根據(jù)該目標(biāo)顏色像素，確定該安全帽的佩戴情況，包括：獲取該感興趣區(qū)域圖像對應(yīng)的總像素；確定該目標(biāo)顏色像素在該總像素中的占比；根據(jù)該占比，確定該安全帽的佩戴情況。