基于改进YOLOv5的复杂环境下安全帽佩戴识别

摘要

由于当前已有方法存在安全帽佩戴识别结果不准确等问题,提出一种基于改进YOLOv5的复杂环境下安全帽佩戴识别方法。优先对施工人员的姿态进行估计,获取施工人员的骨骼点信息。然后采用改进的YOLOv5进一步对人体框架和安全帽框进行识别,获取具体位置信息。根据获取的人体头部关键点确定头部中心区域,采用人体框架较短边确定头部区域边长,得到人体头部区域,实现复杂环境下安全帽佩戴识别。仿真结果表明,所提方法的识别结果精度较高,可以准确对头部区域进行有效定位。