Computer and Modernization ›› 2024, Vol. 0 ›› Issue (12): 66-71.doi: 10.3969/j.issn.1006-2475.2024.12.010

Previous Articles     Next Articles

Safety Helmet Wearing Detection Algorithm for Complex Construction Scenes 

  

  1. (1. School of Computer Science and Technology, Guangdong University of Technology, Guangzhou 510006, China;
    2. School of Automation, Guangdong University of Technology, Guangzhou 510006, China)
  • Online:2024-12-31 Published:2024-12-31

Abstract:  In view of complex background interference and foreign object occlusion in the construction scenes, which reduces the accuracy of helmet wearing detection, we propose a safety helmet wearing detection algorithm for complex construction scenes. This paper improves the YOLOv5 algorithm, adding the Coordinate Attention (CA) mechanism, replacing the first two layers in the backbone network using the Stem Block, applying a Decoupled detection Head (DH) structure with the addition of the Coordinate Attention mechanism. At the same time, an additional large-scale feature extraction layer is added. Results on the helmet dataset show that the improved CADH-YOLOv5 algorithm with a mean detection precision of 91.2% can significantly improve the performance of safety helmet wearing detection for complex construction scenes, which is superior to similar algorithms, and has limited real-time performance.

Key words:  , feature extraction; safety helmet wearing detection; coordinate attention; decoupled detection heads; object detect

CLC Number: