Lightweight safety helmet wearing detection algorithm of improved YOLOv8
DOI:
CSTR:
Author:
Affiliation:

1.School of Information Engineering, Southwest University of Science and Technology,Mianyang 621000, China; 2.Sichuan Provincial Key Laboratory of Robot Technology Used for Special Environment,Mianyang 621000, China

Clc Number:

TN911.73

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In response to the issues of large model size, complex computations, and high resource demands on computational platforms in safety helmet detection models, a lightweight safety helmet detection algorithm called YOLOv8-MBS, based on an improvement of YOLOv8, is proposed. A new lightweight backbone module was first formed by combining MobileNetv3 with SPPF, reducing the algorithm′s parameter and computational load. Moreover, the algorithm′s feature extraction and representation capabilities are enhanced using a weighted bidirectional feature pyramid network, which also reduces the false detection rate. Finally, the SimAM module is incorporated to improve the network′s correlation between positional information and safety helmet features without increasing the computational burden. Experimental results show that compared to the original YOLOv8n network, the improved YOLOv8-MBS maintains high detection accuracy while reducing computation by 35.96%, the number of parameters by 25.63%, and model size by 23.22%, and increasing the frame rate by 12.52 fps. The lightweight nature of the model reduces deployment costs and provides theoretical support for embedded deployment and large-scale applications.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: December 11,2024
  • Published:
Article QR Code