基于YOLOv5和Pose-estimation的工地工人手機使用檢測方法

基本信息

申請?zhí)?/td> CN202011338005.7 申請日 -
公開（公告）號 CN112580778A 公開（公告）日 2021-03-30
申請公布號 CN112580778A 申請公布日 2021-03-30
分類號 G06N3/08(2006.01)I;G06N3/04(2006.01)I;G06T7/00(2017.01)I 分類 計算；推算；計數(shù);
發(fā)明人 仝淑貞;田青;張華;魏煊 申請（專利權(quán)）人 江蘇集萃未來城市應(yīng)用技術(shù)研究所有限公司
代理機構(gòu) 南京科闊知識產(chǎn)權(quán)代理事務(wù)所（普通合伙） 代理人 蘇興建
地址 215131江蘇省蘇州市相城經(jīng)濟技術(shù)開發(fā)區(qū)澄陽街道澄陽路116號陽澄湖國際創(chuàng)業(yè)園3號樓3層301-313房屋
法律狀態(tài) -

摘要

摘要 一種基于YOLOv5和Pose?estimation的工地工人手機使用檢測方法，步驟包括：1)獲取攝像機視頻信流，并截取單幀圖片；2)對圖片進行HSV增強；3)使用coco數(shù)據(jù)集訓(xùn)練YOLOv5網(wǎng)絡(luò)；4)用步驟3)訓(xùn)練得到的YOLOv5神經(jīng)網(wǎng)絡(luò)用來檢測人體，得到人體的檢測框；5)用檢測框?qū)斎雸D像進行裁剪，得到圖像中的人體部分；6)構(gòu)建Pose?estimation網(wǎng)絡(luò)；8)將YOLOv5神經(jīng)網(wǎng)絡(luò)檢測出的人體輸入Pose?estimation網(wǎng)絡(luò)，輸出人體的手部ROI；9)將手部ROI輸入到ResNet?50分類網(wǎng)絡(luò)，預(yù)測手部區(qū)域中是否含有手機，得到結(jié)果。本方法大大降低人力成本并提高監(jiān)督效果。??