基于YOLOv8的安全帽检测

前言

        建筑工人头部伤害是造成建筑伤亡事故的重要原因。佩戴安全帽是防止建筑工人发生脑部外伤事故的有效措施,而在实际工作中工人未佩戴安全帽的不安全行为时有发生。因此,对施工现场建筑工人佩戴安全帽自动实时检测进行探究,将为深入认知和主动预防安全事故提供新的视角。然而,传统的施工现场具有安全管理水平低下、管理范围小、主要依靠安全管理人员的主观监测并且时效性差、不能全程监控等一系列问题。 

1.YOLOv8介绍

  Ultralytics YOLOv8是Ultralytics公司开发的YOLO目标检测和图像分割模型的最新版本。YOLOv8是一种尖端的、最先进的（SOTA）模型，它建立在先前YOLO成功基础上，并引入了新功能和改进，以进一步提升性能和灵活性。它可以在大型数据集上进行训练，并且能够在各种硬件平台上运行，从CPU到GPU。
 

Yolo采用卷积网络来提取特征，然后使用全连接层来得到预测值。网络结构参考GooLeNet模型，包含24个卷积层和2个全连接层。

Loss 计算

Loss 计算过程包括 2 个部分： 正负样本分配策略和 Loss 计算。

 2.工作原理

1）判断工人存在的区域，使用人脸检测模块对人脸进行标记；

采用YOLOv8算法，其目标检测框架可以实现对多类目标物体的检测。

2）提取工人头部子区域；

统一在各区域的中部靠上方部分提取，将提取区域固定为正方，将分离出来的头部区域再输入到之后的网络中作后续分析。

3）对提取到的图像采用二分类方法，判断工人是否佩戴安全帽。

用二分类法完成对是否佩戴了安全帽的情况进行判断。

3.搭建YOLOv8运行环境

1.下载yolov8代码
github：YOLOv8-github

gitee:YOLOv8-gitee

推荐使用国内的gitee

2.创建conda虚拟环境
如果没有安装conda可以搜索一下conda配置教程，按照流程安装好conda，还要下载好符合自己电脑版本的CUDA 后续会用。

3.打开conda窗口 进入到安装的YOLOv8界面
1.创建新的虚拟环境
conda create -n y8 python=3.8
2.输入下面命令查看是否创建成功
 conda env list 
3.激活进入环境
conda activate y8
4.安装配置文件

首先先把pip的源换到国内aliyun镜像，下载速度提高很多
pip config set install.trusted-host mirrors.aliyun.com

首先conda环境cd进入主文件夹下࿰