测试科学与仪器2023，Vol.14Issue(4)：463-472,10.DOI:10.3969/j.issn.1674-8042.2023.04.009
改进YOLOv5适应安全帽佩戴与口罩佩戴检测应用的算法
An improved algorithm for adapting YOLOv5 to helmet wearing and mask wearing detection applications
摘要
Abstract
In order to achieve more efficient detection of wearing helmets and masks in natural scenes, an improved algorithm model YOLOv5+ is proposed based on the deep learning algorithm YOLOv5. For target detection tasks, small targets are usually detected on a large feature map. Considering that most of the detected objects are small-scale targets. Therefore, when the input image size is 640×640 pixels by default, a feature map of size 160×160 pixels is added to the detection layer of the original algorithm, and complete intersection over union (CIoU) is selected as the loss function to achieve more effective detection of helmet wearing and mask wearing. The experimental results show that the mean average precision (mAP-50) of the YOLOv5+ network model reaches 93.8％ and 92.3％ on the helmet-wearing and mask-wearing datasets, respectively, which is both improved compared to the precision of the original algorithm. This method not only meets the speed requirement of real-time detection, but also improves the precision of detection.关键词
YOLOv5/CIoU/安全帽佩戴检测/口罩佩戴检测/适应小目标检测Key words
YOLOv5/CIoU/helmet wear detection/mask wear detection/adaptation to small target detection引用本文复制引用
张又元,杨桂芹,刁广超,孙存威,王小鹏..改进YOLOv5适应安全帽佩戴与口罩佩戴检测应用的算法[J].测试科学与仪器,2023,14(4):463-472,10.基金项目
National Natural Science Foundation of China(No.61761027) （No.61761027）
