基于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分类网络,预测手部区域中是否含有手机,得到结果。本方法大大降低人力成本并提高监督效果。 |
