基于YOLOv5和Pose-estimation的工地工人手机使用检测方法

基本信息

申请号 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 分类 计算;推算;计数;
发明人 仝淑贞;田青;张华;魏煊 申请(专利权)人 江苏集萃未来城市应用技术研究所有限公司
代理机构 南京科阔知识产权代理事务所(普通合伙) 代理人 苏兴建
地址 215131江苏省苏州市相城经济技术开发区澄阳街道澄阳路116号阳澄湖国际创业园3号楼3层301-313房屋
法律状态 -

摘要

摘要 一种基于YOLOv5和Pose‑estimation的工地工人手机使用检测方法,步骤包括:1)获取摄像机视频信流,并截取单帧图片;2)对图片进行HSV增强;3)使用coco数据集训练YOLOv5网络;4)用步骤3)训练得到的YOLOv5神经网络用来检测人体,得到人体的检测框;5)用检测框对输入图像进行裁剪,得到图像中的人体部分;6)构建Pose‑estimation网络;8)将YOLOv5神经网络检测出的人体输入Pose‑estimation网络,输出人体的手部ROI;9)将手部ROI输入到ResNet‑50分类网络,预测手部区域中是否含有手机,得到结果。本方法大大降低人力成本并提高监督效果。