Open Access iconOpen Access

ARTICLE

crossmark

Real-time Safety Helmet-wearing Detection Based on Improved YOLOv5

Yanman Li1, Jun Zhang1, Yang Hu1, Yingnan Zhao2,*, Yi Cao3

1 NARI Nanjing Control System Ltd., Nanjing, 210032, China
2 School of Computer, Nanjing University of Information Science & Technology, Nanjing, 210044, China
3 Department of Electrical and computer Engineering, University of Windsor, Windsor, N9B 3P4, Canada

* Corresponding Author: Yingnan Zhao. Email: email

Computer Systems Science and Engineering 2022, 43(3), 1219-1230. https://doi.org/10.32604/csse.2022.028224

Abstract

Safety helmet-wearing detection is an essential part of the intelligent monitoring system. To improve the speed and accuracy of detection, especially small targets and occluded objects, it presents a novel and efficient detector model. The underlying core algorithm of this model adopts the YOLOv5 (You Only Look Once version 5) network with the best comprehensive detection performance. It is improved by adding an attention mechanism, a CIoU (Complete Intersection Over Union) Loss function, and the Mish activation function. First, it applies the attention mechanism in the feature extraction. The network can learn the weight of each channel independently and enhance the information dissemination between features. Second, it adopts CIoU loss function to achieve accurate bounding box regression. Third, it utilizes Mish activation function to improve detection accuracy and generalization ability. It builds a safety helmet-wearing detection data set containing more than 10,000 images collected from the Internet for preprocessing. On the self-made helmet wearing test data set, the average accuracy of the helmet detection of the proposed algorithm is 96.7%, which is 1.9% higher than that of the YOLOv5 algorithm. It meets the accuracy requirements of the helmet-wearing detection under construction scenarios.

Keywords


Cite This Article

APA Style
Li, Y., Zhang, J., Hu, Y., Zhao, Y., Cao, Y. (2022). Real-time safety helmet-wearing detection based on improved yolov5. Computer Systems Science and Engineering, 43(3), 1219-1230. https://doi.org/10.32604/csse.2022.028224
Vancouver Style
Li Y, Zhang J, Hu Y, Zhao Y, Cao Y. Real-time safety helmet-wearing detection based on improved yolov5. Comput Syst Sci Eng. 2022;43(3):1219-1230 https://doi.org/10.32604/csse.2022.028224
IEEE Style
Y. Li, J. Zhang, Y. Hu, Y. Zhao, and Y. Cao, “Real-time Safety Helmet-wearing Detection Based on Improved YOLOv5,” Comput. Syst. Sci. Eng., vol. 43, no. 3, pp. 1219-1230, 2022. https://doi.org/10.32604/csse.2022.028224



cc Copyright © 2022 The Author(s). Published by Tech Science Press.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 2864

    View

  • 1759

    Download

  • 2

    Like

Share Link