基于YOLOv5的电动车安全头盔佩戴检测系统

本系统是基于YOLOv5的实现头盔识别的C/S架构。总体方案设计包含模型设计和C/S架构设计。

1.模型设计

“头盔佩戴检测”问题的实现要求为：

（1）通过图片或视频判断是否佩戴头盔。

（2）用户希望得到的是，定位出佩戴头盔的目标位置,并判断是否佩戴头盔。

（3）头盔的所有角度所有方向都可以识别，应对突发情况，提高该模型的监测精准度。

基于以上分析，项目实现流程包括数据集预处理、模型设计及训练。且这里注重头盔“佩戴”状态，需要的数据是有“佩戴”头盔或未“佩戴”头盔的视频。如图1所示：

图1  佩戴状态

其中红色的bounding box是未佩戴头盔的目标，绿色的bounding box是佩戴头盔的目标，这样可以更精准、更直接的确定是否佩戴的状态，不过需要覆盖更多场景的目标数据。

主要采用Yolov5算