正文

安全帽是作业场所作业时头部防护所用的头部防护用品，它对使用者的头部在受坠落物或小型飞溅物体等其他因素引起的伤害起到防护作用。近年来，因不佩戴安全帽、不规范佩戴安全帽等原因导致的安全生产事故屡禁不止，事故发生背后的影响是巨大的，不仅为家人带来巨大的伤痛，也为企业的利益带来巨大的损失。而如何使员工规范佩戴安全帽、保障员工和企业双方利益，成为了一直以来各方坚持不懈想要实现的目标。因此，研究佩戴安全帽的监测算法，具有重大的意义和广泛的应用价值。

本篇博客，将手把手教你搭建一个基于YOLOv5的佩戴安全帽目标检测识别方法。目前，基于YOLOv5s的目标检测的佩戴安全帽识别方法的平均精度平均值mAP_0.5=0.93，mAP_0.5:0.95=0.63，基本满足业务的性能需求。

另外，为了能部署在手机Android平台上，本人对YOLOv5s进行了模型轻量化，开发了一个轻量级的版本yolov5s05_416和yolov5s05_320，在普通Android手机上可以达到实时的检测和识别效果，CPU(4线程)约30ms左右，GPU约25ms左右，基本满足业务的性能需求。

先展示一下Python版本佩戴安全帽检测和识别Demo效果（head表示(头部)未佩戴安全帽，helmet表示佩戴了安全帽）：

【源码下载】 YOLOv5实现佩戴安全帽检测和识别(含佩戴安全帽数据集+训练代码)

【尊重原创，转载请注明出处】https://panjinquan.blog.csdn.net/article/details/127250780

还有Android版本的佩戴安全帽检测效果：

2. 佩戴安全帽检测和识别的方法

（1）基于目标检测的佩戴安全帽识别方法

基于目标检测的佩戴安全帽识别方法，一步到位，把佩戴安全帽类别直接当成多个目标检测的类别进行训练。

该方案采用one-stage的方法，直接端到端训练，任务简单，速度较快；
新增类别或者数据，需要人工拉框标注佩戴安全帽的框，成本较大
部署简单

（2）基于头部检测+佩戴安全帽分类识别方法

该方法，先训练一个通用的头部检测模型(不区分是否佩戴安全帽)，然后外扩检测框并裁剪头部区域，再训练一个佩戴安全帽分类器，完成对佩戴安全帽的分类识别。当然一个更加简单的方法是，可以将人脸检测器当做头部检测器，即将人脸检测框外扩2倍左右，作为头部检测框的区域。只是这样做，若人背向，人脸检测器一般效果比较差，无法识别是否佩戴了安全帽。

该方案采用two-stage方法，可针对性分别提高检测模型和分类模型的性能
头部检测模型不区分是否佩戴安全帽，只检测头部检测框，检测精度较高，
佩戴安全帽分类模型可以做到很轻量
由于采用two-stage方法进行检测-识别，因此速度相对较慢

考虑到佩戴安全帽检测和识别的任务比较简单，因此本博客将采用“基于目标检测的佩戴安全帽识别方法”。目标检测的的方法较多，比如Faster-RCNN，YOLO系列，SSD等均可以采用，本博客将采用YOLOv5进行佩戴安全帽目标检测模型训练。

3. 佩戴安全帽数据集说明

（1）佩戴安全帽数据集

目前收集了2W+的佩戴安全帽数据集：

Helmet-Asian亚洲人佩戴安全帽数据集：总共有19000+图片，VOC的XML数据格式，可直接用于目标检测模型训练。
Helmet-Europe欧洲人佩戴安全帽数据集：总共有3000+图片，VOC的XML数据格式，可直接用于目标检测模型训练。

关于佩戴安全帽数据集说明，详见另一篇博客说明：佩戴安全帽数据集使用说明和下载_PKing666666的博客-CSDN博客

（2）自定义数据集

如果需要增/删类别数据进行训练，或者需要自定数据集进行训练，可参考如下步骤：

采集图片，建议不少于200张图片
使用Labelme等标注工具，对目标进行拉框标注：labelme工具：GitHub - wkentaro/labelme: Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).
将标注格式转换为VOC数据格式，参考工具：labelme/labelme2voc.py at main · wkentaro/labelme · GitHub
生成训练集train.txt和验证集val.txt文件列表
修改engine/configs/voc_local.yaml的train和val的数据路径
重新开始训练

4. 基于YOLOv5的佩戴安全帽模型训练

（1）YOLOv5说明

训练Pipeline采用YOLOv5: https://github.com/ultralytics/yolov5 , 原始代码训练需要转换为YOLO的格式，不支持VOC的数据格式。为了适配VOC数据，本人新增了LoadVOCImagesAndLabels用于解析VOC数据集进行训练。另外，为了方便测试，还增加demo.py文件，可支持对图片,视频和摄像头的测试。

Python依赖环境，使用pip安装即可：


matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow
PyYAML>=5.3.1
scipy>=1.4.1
torch>=1.7.0
torchvision>=0.8.1
tqdm>=4.41.0
tensorboard>=2.4.1
seaborn>=0.11.0
pandas
thop  # FLOPs computation
pybaseutils

（2）准备Train和Test数据

下载Helmet-Asian和Helmet-Europe佩戴安全帽数据集，总共约2W+的图片

Helmet-Asian	Helmet-Europe

（3）配置数据文件

修改训练和测试数据的路径：engine/configs/voc_local.yaml

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
# 数据路径
path: ""  # dataset root dir
train:
  - "/path/to/helmet/Helmet-Asian/trainval.txt"
  - "/path/to/helmet/Helmet-Europe/trainval.txt"

val:
  - "/path/to/helmet/Helmet-Asian/test.txt"

test:  # test images (optional)
data_type: voc

# Classes
nc: 2  # number of classes
names:  'head': 0, "helmet": 1

（4）配置模型文件

官方YOLOv5给出了YOLOv5l,YOLOv5m,YOLOv5s等模型。考虑到手机端CPU/GPU性能比较弱鸡，直接部署yolov5s运行速度十分慢。所以本人在yolov5s基础上进行模型轻量化处理，即将yolov5s的模型的channels通道数全部都减少一半，并且模型输入由原来的640×640降低到416×416或者320×320，该轻量化的模型我称之为yolov5s05。从性能来看，yolov5s05比yolov5s快5多倍，而mAP下降了9%（0.93→0.84），对于手机端，这精度勉强可以接受。

下面是yolov5s05和yolov5s的参数量和计算量对比：

模型	input-size	params(M)	GFLOPs	mAP0.5
yolov5s	640×640	7.2	16.5	0.93067
yolov5s05	416×416	1.7	1.8	0.87051
yolov5s05	320×320	1.7	1.1	0.84477

（5）重新聚类Anchor（可选）

官方yolov5s的Anchor是基于COCO数据集进行聚类获得（详见models/yolov5s.yaml文件）

对于yolov5s05的Anchor，由于输入大小640缩小到320，其对应的Anchor也应该缩小一倍：

一点建议：

官方yolov5s的Anchor是基于COCO数据集进行聚类获得，不同数据集需要做适当的调整，其最优Anchor建议重新进行聚类。
当然你要是觉得麻烦就跳过，不需要重新聚类Anchor，这个影响不是特别大。如果你需要重新聚类，请参考engine/kmeans_anchor/demo.py文件

（6）开始训练

整套训练代码非常简单操作，用户只需要将相同类别的数据放在同一个目录下，并填写好对应的数据路径，即可开始训练了。

修改训练超参文件： data/hyps/hyp.scratch-v1.yaml （可以修改训练学习率，数据增强等方式，使用默认即可）
编辑train.sh脚本训练，训练yolov5s或轻量化版本yolov5s05_416或者yolov5s05_320 (选择其中一个训练即可)

#!/usr/bin/env bash

#--------------训练yolov5s--------------
# 输出项目名称路径
project="runs/yolov5s_640"
# 训练和测试数据的路径
data="engine/configs/voc_local.yaml"
# YOLOv5模型配置文件
cfg="yolov5s.yaml"
# 训练超参数文件
hyp="data/hyps/hyp.scratch-v1.yaml"
# 预训练文件
weights="engine/pretrained/yolov5s.pt"
python train.py --data $data --cfg $cfg --hyp $hyp --weights $weights --batch-size 64 --imgsz 640 --workers 12 --project $project


#--------------训练轻量化版本yolov5s05_416--------------
# 输出项目名称路径
project="runs/yolov5s05_416"
# 训练和测试数据的路径
data="engine/configs/voc_local.yaml"
# YOLOv5模型配置文件
cfg="yolov5s05_416.yaml"
# 训练超参数文件
hyp="data/hyps/hyp.scratch-v1.yaml"
# 预训练文件
weights="engine/pretrained/yolov5s.pt"
python train.py --data $data --cfg $cfg --hyp $hyp --weights $weights --batch-size 64 --imgsz 416 --workers 12 --project $project


#--------------训练轻量化版本yolov5s05_320--------------
# 输出项目名称路径
project="runs/yolov5s05_320"
# 训练和测试数据的路径
data="engine/configs/voc_local.yaml"
# YOLOv5模型配置文件
cfg="yolov5s05_320.yaml"
# 训练超参数文件
hyp="data/hyps/hyp.scratch-v1.yaml"
# 预训练文件
weights="engine/pretrained/yolov5s.pt"
python train.py --data $data --cfg $cfg --hyp $hyp --weights $weights --batch-size 64 --imgsz 320 --workers 12 --project $project

开始训练： bash train.sh

训练完成，可以得到yolov5s佩戴安全帽识别的mAP指标大约mAP_0.5=0.93；而

yolov5s05_416 mAP_0.5=0.87左右；yolov5s05_320 mAP_0.5=0.84左右

（7）可视化训练过程

训练过程可视化工具是使用Tensorboard，使用方法：

# 基本方法
tensorboard --logdir=path/to/log/
# 例如
tensorboard --logdir ./runs/yolov5s_640

当然，在输出目录，也保存很多性能指标的图片

这是训练epoch的可视化图，可以看到mAP随着Epoch训练，逐渐提高

这是每个类别的F1-Score分数

这是模型的PR曲线

这是混淆矩阵：

5. Python版本佩戴安全帽检测效果

demo.py文件用于推理和测试模型的效果，填写好配置文件，模型文件以及测试图片即可运行测试了

测试图片

# 测试图片
image_dir='data/helmet-test' # 测试图片的目录
weights="runs/model/yolov5s_640/weights/best.pt" # 模型文件
out_dir="runs/helmet-result" # 保存检测结果
python demo.py --image_dir $image_dir --weights $weights --out_dir $out_dir

测试视频文件

# 测试视频文件
video_file="data/helmet-test.mp4" # path/to/video.mp4 测试视频文件，如*.mp4,*.avi等
weights="runs/model/yolov5s_640/weights/best.pt" # 模型文件
out_dir="runs/helmet-result" # 保存检测结果
python demo.py --video_file $video_file --weights $weights --out_dir $out_dir

测试摄像头

# 测试摄像头
video_file=0 # 测试摄像头ID
weights="runs/model/yolov5s_640/weights/best.pt" # 模型文件
out_dir="runs/helmet-result" # 保存检测结果
python demo.py --video_file $video_file --weights $weights --out_dir $out_dir

先展示一下Python版本佩戴安全帽检测和识别Demo效果（head表示(头部)未佩戴安全帽，helmet表示佩戴了安全帽）：

如果想进一步提高模型的性能，可以尝试：

增加训练的样本数据：目前只有2W+的数据量，建议根据自己的业务场景，采集相关数据，提高模型泛化能力
使用参数量更大的模型：本教程使用的YOLOv5s，其参数量才7.2M，而YOLOv5x的参数量有86.7M，理论上其精度更高，但推理速度也较慢。
尝试不同数据增强的组合进行训练

6. Android版本佩戴安全帽检测效果（即将发布）

已经完成Android版本的佩戴安全帽目标检测识别模型算法开发，APP在普通Android手机上可以达到实时的检测和识别效果，CPU(4线程)约30ms左右，GPU约25ms左右，基本满足业务的性能需求。

Android版本佩戴安全帽识别：https://panjinquan.blog.csdn.net/article/details/127345231

Android Demo效果：

【Android APP体验】https://download.csdn.net/download/guyuealian/86768318

7.项目源码下载

整套项目源码内容包含佩戴安全帽数据集 + YOLOv5训练代码和测试代码

【源码下载】YOLOv5实现佩戴安全帽检测和识别(含佩戴安全帽数据集+训练代码)

（1）佩戴安全帽数据集,总共2W+图片

Helmet-Asian亚洲人佩戴安全帽数据集：总共有19000+图片，VOC的XML数据格式，可直接用于目标检测模型训练。
Helmet-Europe欧洲人佩戴安全帽数据集：总共有3000+图片，VOC的XML数据格式，可直接用于目标检测模型训练。

关于数据详细说明：佩戴安全帽数据集使用说明和下载_PKing666666的博客-CSDN博客

（2）YOLOv5训练代码和测试代码（Pytorch）

整套YOLOv5项目工程的训练代码和测试代码
支持高精度版本yolov5s训练和测试
支持轻量化版本yolov5s05_320和yolov5s05_416训练和测试，可以用于在Android手机实时检测
根据本篇博文说明，简单配置即可开始训练

android实现佩戴安全帽检测和识别(含训练代码+android源码+数据集)(代码片段)

Android实现佩戴安全帽检测和识别(含训练代码+Android源码+数据集)目录 Android实现佩戴安全帽检测和识别(含训练代码+Android源码+数据集)1.前言2.佩戴安全帽检测和识别的方法（1）基于目标检测的佩戴安全帽识... 查看详情

基于深度学习的安全帽检测系统（yolov5清新界面版，python代码）(代码片段)

安全帽检测系统用于自动化监测安全帽佩戴情况，在需要佩戴安全帽的场合自动安全提醒，实现图片、视频和摄像头等多种形式监测。在介绍算法原理的同时，给出Python的实现代码、训练数据集，以及PyQt的UI界面。安全帽检测系... 查看详情

保姆教程|yolov5在建筑工地中安全帽佩戴检测的应用

教程来自：https://github.com/PeterH0323/Smart_Construction编辑:AI深度前沿视线一、YOLOv5训练自己数据集教程 1.1创建自己的数据集配置文件 1.2创建每个图片对应的标签文件 1.3文件放置规范 1.4聚类得出先验框（可选... 查看详情

基于深度学习的安全帽检测系统（yolov5清新界面版，python代码）(代码片段)

摘要：安全帽检测系统用于自动化监测安全帽佩戴情况，在需要佩戴安全帽的场合自动安全提醒，实现图片、视频和摄像头等多种形式监测。在介绍算法原理的同时，给出Python的实现代码、训练数据集，以及Py... 查看详情

pytorch深度学习实战|基于yolov3的安全帽佩戴检测

本期将提供一个利用深度学习检测是否佩戴安全帽的案例，从而展示计算机视觉中的目标识别问题的一般流程。目标检测是基于图片分类的计算机视觉任务，既包含了分类，又包含了定位。给出一张图片，目标检测系统要能够识... 查看详情

寒假学习进度

...今天毕业设计没有进行新功能的实现，改进了一下原有的安全帽和未佩戴安全帽的人数统计功能，使之更加符合现实中的工程要求。原有的人数统计是统计每一帧中的佩戴安全帽的人和未佩戴安全帽的人，在每一帧中实时更新显... 查看详情

开源一个安全帽佩戴检测数据集及预训练模型

转载自开源一个安全帽佩戴检测数据集及预训练模型-supersayajin-博客园本文开源了一个安全帽佩戴检测数据集及预训练模型，该项目已上传至github，点此链接，感觉有帮助的话请点star。同时简要介绍下实践上如何完... 查看详情

tsingsee青犀视频开发安全帽智能检测识别经验分享

安全帽佩戴智能检测技术实际就是检测施工人员是否佩戴安全帽。在当前AI人工智能和大数据兴起与飞速发展的阶段，安全帽智能检测技术已经不是什么新鲜事儿了，大家接触的安全帽智能识别与检测也已经逐渐趋于成熟... 查看详情

口罩识别/安全帽识别/区域入侵检测/ai智能分析网关新增网口配置模块

...物体等进行检测、跟踪与抓拍，支持口罩佩戴检测、安全帽佩戴检测、人体检测、区域入侵检测及可拓展多种AI检测算法。在功能上，我们也在持续开发和完善相关功能查看详情

佩戴安全帽数据集使用说明和下载

佩戴安全帽数据集使用说明和下载安全帽是作业场所作业时头部防护所用的头部防护用品，它对使用者的头部在受坠落物或小型飞溅物体等其他因素引起的伤害起到防护作用。一般由帽壳、帽衬及配件等组成。网上已经开... 查看详情

人脸识别/安全帽识别ai智能分析网关微信端告警推送如何配置模板消息？

...物体等进行检测、跟踪与抓拍，支持口罩佩戴检测、安全帽佩戴检测、人体检测、区域入侵检测等，可支持拓展多种AI检测算法。其中，升级版还可支持人脸识别、车辆识别、车牌识别、烟火检测、电动车检测等，... 查看详情

智能分析网关ai安全帽识别技术在工地场景中的应用及意义

安全帽是建筑业、制造业等工业生产中重要的劳保工具，应用十分广泛且十分重要。但是在实际场景中，比如建筑工地或工厂流水线上，依然有很多工人忽视安全帽的重要性，同时，由于企业的监督不到位ÿ... 查看详情

安全帽检测

安全帽检测算法介绍：安全帽检测算法主要用工厂，建筑工地的入口，工作区域，可以进入作业区域的人员进行自动检测。若检测到人员未佩戴安全帽，可立即告警，该算法极大地提升了作业区域的管控效... 查看详情

智能驾驶车牌检测和识别《yolov5实现车牌检测（含车牌检测数据集和训练代码）》(代码片段)

智能驾驶车牌检测和识别（二）《YOLOv5实现车牌检测（含车牌检测数据集和训练代码）》目录智能驾驶车牌检测和识别（二）《YOLOv5实现车牌检测（含车牌检测数据集和训练代码）》1.前言2.车牌检... 查看详情

智能驾驶车牌检测和识别《yolov5实现车牌检测（含车牌检测数据集和训练代码）》(代码片段)

基于python实现的口罩佩戴检测(代码片段)

口罩佩戴检测一题目背景1.1实验介绍今年一场席卷全球的新型冠状病毒给人们带来了沉重的生命财产的损失。有效防御这种传染病毒的方法就是积极佩戴口罩。我国对此也采取了严肃的措施，在公共场合要求人们必须佩戴口... 查看详情

五分钟快速搭建一个实时人脸口罩检测系统(opencv+paddlehub含源码)(代码片段)

导读本文主要介绍如何使用OpenCV和PaddleHub实现一个实时人脸口罩检测系统。（公众号：OpenCV与AI深度学习）背景介绍从19年疫情爆发到现在，佩戴口罩对大家来说已是常态。应运而生的就有了很多相关应用࿰... 查看详情