Researching | Improved YOLOv4 Helmet Detection Algorithm Under Complex Scenarios

Laser & Optoelectronics Progress, Volume. 60, Issue 12, 1210011(2023)

Improved YOLOv4 Helmet Detection Algorithm Under Complex Scenarios

Guobo Xie¹, Jingjing Tang¹, Zhiyi Lin^1、*, Xiaofeng Zheng¹, and Ming Fang²

Author Affiliations

¹School of Computer Science and Technology, Guangdong University of Technology, Guangzhou 510006, Guangdong, China

²Transmission Branch of Yunnan Power Grid Co., Ltd., Kunming 650011, Yunnan, China

show less

Abstract Get PDF(in Chinese)

Figures&Tables (10)

References (21)

Paper Information

Figures & Tables(10)

Fig. 1. YOLOv4 network structure

Download full size

Fig. 2. SMD-YOLOv4 network structure

Download full size

Fig. 3. SE-Net structure

Download full size

Fig. 4. Multi-scale information fusion detection network

Download full size

Fig. 5. DenseASPP network structure

Download full size

Fig. 6. SMD-YOLOv4 model training situation. (a) Loss; (b) mAP

Download full size

Fig. 7. Detection effect comparison of different models

Download full size

Fig. 8. Comparison of detection effect with other deep learning target detection algorithms

Download full size

Table 1. Comparison of the results of ablation experiments
View table
Table 1. Comparison of the results of ablation experiments
YOLOv4 SE Multi-scale Dw DenseASPP P /% R /% mAP /% T /ms
√ 92.30 89.76 94.15 24.7
√ √ 93.72 88.78 94.74 34.6
√ √ √ 94.80 89.60 95.41 51.4
√ √ √ √ 94.51 89.40 95.23 28.5
√ √ √ √ √ 95.19 93.00 97.34 47.1

Table 2. Performance comparison with other deep learning target detection algorithms
View table
Table 2. Performance comparison with other deep learning target detection algorithms
Model P /% R /% mAP /% T /ms
Faster R-CNN 57.31 75.31 70.93 40.1
SSD 93.79 76.67 90.90 12.9
YOLOv5 94.81 86.16 94.09 23.7
YOLOx 94.51 91.33 95.85 24.6
SMD-YOLOv4 95.19 93.00 97.34 47.1

Tools

Get Citation

Copy Citation Text

Guobo Xie, Jingjing Tang, Zhiyi Lin, Xiaofeng Zheng, Ming Fang. Improved YOLOv4 Helmet Detection Algorithm Under Complex Scenarios[J]. Laser & Optoelectronics Progress, 2023, 60(12): 1210011

Download Citation

EndNote(RIS)BibTex Plain Text

Set citation alerts for article

Save article for my favorites

Paper Information

Category: Image Processing

Received: Apr. 22, 2022

Accepted: Jun. 16, 2022

Published Online: May. 23, 2023

The Author Email: Zhiyi Lin (lzy291@gdut.edu.cn)

DOI:10.3788/LOP221388

Topics

laser devices and laser physics

Lasers and Laser Optics

laser manufacturing

Instrumentation, Measurement and Metrology