基于Yolov5的目标检测（以口罩检测为例）

Vellichor_00x

已于 2022-12-30 10:34:42 修改

阅读量2k

点赞数 8

分类专栏：深度学习项目文章标签：目标检测深度学习计算机视觉

于 2022-12-29 23:54:14 首次发布

本文链接：https://blog.csdn.net/m0_62577716/article/details/128489315

版权

深度学习项目专栏收录该内容

2 篇文章 0 订阅

订阅专栏

口罩数据集：GitHub - X-zhangyang/Real-World-Masked-Face-Dataset: Real-World Masked Face Dataset，口罩人脸数据集

yolov5 源码：GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

一.深度学习目标检测算法分类

1. 基于区域提取的两阶段目标检测

1.1 SPP-Net

1.2 R-CNN

1.3 FAST R-CNN

1.4 FASTER-CNN

2. 基于回归的单阶段为目标检测

2.1 YOLO系列(YOLO,YOLOv1,2,3,4,5,7)

2.2 OverFeat

二.YOLO系列目标检测算法介绍

YOLOv1：YOLOv1是利用整张图作为网络的输入，直接在输出层回归bounding box的位置和bounding box所属的类别。
YOLOv2：YOLOv2相较v1版本，预测更准确，速度更快，识别对象更多这三个方面进行了改进
YOLOv3：YOLOv3的Bounding Box由YOLOv2又做出了更好的改进
YOLOv4：YOLOv4结合了大量前人研究技术，加以组合并进行适当创新的算法，实现了速度和精度的完美平衡。
YOLOv5：YOLOv5在YOLOv4算法的基础上做了进一步的改进，检测性能得到进一步的提升。在COCO数据集上面的测试效果较好。

三 .Yolov5介绍

3.1 YOLOv5网络模型主要包括：Yolov5s、Yolov5m、Yolov5l、Yolov5x

3.2 Yolov5s网络为Yolov5系列中深度最小，特征图的宽度最小的网络。而后面的3种则是在此基础上不断加深，不断加宽，速度的消耗也在不断增加

3.3 YOLOv5主要部分

四、基于yolov5的人脸口罩识别检测

4.1 受新型冠状病毒的影响，人脸口罩佩戴检测成为了当前急需的应用，而YOLOv5作为目前检测能力较强的目标检测技术，因此，结合本次专业认识实习以及所学知识，我使用了YOLOv5来实现人脸口罩佩戴的检测。实践的内容包括：运用网络爬虫爬取一定量的数据集，运用MakeSense对数据集进行打标签，然后在云服务器上利用YOLOv5网络模型训练数据集得到最佳模型，最后利用其实现人脸口罩佩戴的一定精度的检测。

4.2 实现过程：

4.2.1：获取并准备数据集（此数据集较小，需要大数据集的可参考我上传的资源或者开源数据集）

Part2：包括100张（不佩戴口罩）

Part3：包括100张（佩戴口罩）

4.2.2：利用MakeSense进行逐个打标签（也可以使用labelme进行打标签）

MaskSense上手快，使用较为容易

链接: Make Sense

4.2.3. 为后期的训练

需要将数据集mydata首先划分为两个部分(一个存放图片，一个存放标签)

1.image, 2.labels

并划分为训练集，测试集与验证集

1.test, 2.train, 3train_2

4.2.4 文件目录介绍

data：主要是存放一些超参数的配置文件,是用来配置训练集和测试集还有验证集的路径，其中还包括目标检测的种类数和种类的名称；还有官方提供测试的图片。训练自己的数据集需要修改其中的yaml文件

models：里面主要是一些网络构建的配置文件和函数，其中包含了该项目的四个不同的版本，分别为是s、m、l、x

utils：存放的是工具类的函数，里面有loss函数，metrics函数，plots函数等等

weights：放置训练好的权重参数

detect.py：利用训练好的权重参数进行目标检测，可以进行图像、视频和摄像头的检测

train.py：训练自己的数据集的函数

test.py：测试训练的结果的函数