目标检测论文解读复现之十九：基于YOLOv5网络模型的人员口罩佩戴实时检测

前言

 

此前出了目标改进算法专栏，但是对于应用于什么场景，需要什么改进方法对应与自己的应用场景有效果，并且多少改进点能发什么水平的文章，为解决大家的困惑，此系列文章旨在给大家解读最新目标检测算法论文，帮助大家解答疑惑。解读的系列文章，本人已进行创新点代码复现，有需要的朋友可关注私信我。

一、摘要

 近年来，随着硬件算力的提升和人工智能算法的创新发展，使得深度学习算法在目标检测方面有着广泛的应用。针对现有人工方式查看人员口罩佩戴情况的不足，提出了一种基于深度学习YOLOv5算法实现对口罩佩戴情况的实时检测。算法首先将数据集进行归一化处理，再将数据接入YOLOv5网络进行迭代训练，并将最优权重数据保存用作测试集测试，算法通过tensorboard可视化显示训练和测试结果。实验结果表明，所提算法检测的准确性高，实时性强，满足实际使用需求。

二、网络模型及核心创新点

1. Input端

2. Backbone

3. Neck

4. Prediction

三、应用数据集

实验从线上收集了5 000张人员口罩佩戴图像，制作训练和测试数据集，数据集包括了3种类别，分别为mask、none、poor。

四、实验效果（部分展示）

1. 模型训练完成后，将测试数据集输入至模型中，测试结果如下图所示。

2. 下表给出了本文算法与参考文献中其他3种算法测试性能对比，本文算法的准确率为92.4%，分别比文献中的3种算法提高了6%、3.4%、4.7%;检测速度为140帧/s，相比于文献［8］中的算法提高了近8倍的速度，这得益于YOLOv5较小的权重数据。

、实验结论

综上所述，为实现检测的准确性和实时性，并将人工从繁杂的检测工作中解放出来，提出一种基于YOLOv5网络模型的人员口罩佩戴实时检测算法。该方法经实验测试，准确率高且能达到视频图像实时性要求，证明本文方法具有一定的优势。

六、投稿期刊介绍

注：论文原文出自谈世磊，别雄波 ，卢功林，谈小虎；基于YOLOv5网络模型的人员口罩佩戴实时检测；《激光杂志》。

解读的系列文章，本人已进行创新点代码复现，有需要的朋友可关注下面公众号，私信我。