Efficient and Lightweight Neural Network for Hard Hat Detection

Chenxi He,Shengbo Tan,Jing Zhao,Daji Ergu,Fangyao Liu,Bo Ma,Jianjun Li

DOI: https://doi.org/10.3390/electronics13132507

IF: 2.9

2024-06-27

Electronics

Abstract:Electric power operation, as one of the key fields in the world, faces particularly prominent safety issues. Ensuring the safety of operators has become the most fundamental requirement in power operation. However, there are some safety hazards in power construction. These hazards are mainly due to weak safety awareness among staff and the failure to standardize the wearing of safety helmets. In order to effectively address this situation, technical means such as video surveillance technology and computer vision technology can be utilized to monitor whether staff are wearing helmets and provide timely feedback. Such measures will greatly enhance the safety level of power operation. This paper proposes an improved lightweight helmet detection algorithm named YOLO-M3C. The algorithm first replaces the YOLOv5s backbone network with MobileNetV3, successfully reducing the model size from 13.7 MB to 10.2 MB, thereby increasing the model's detection speed from 42.0 frames per second to 55.6 frames per second. Then, the CA attention mechanism is introduced into the backbone network to enhance the feature extraction capability of the model. Finally, in order to further improve the detection recall rate and accuracy of the model, a knowledge distillation of the model was carried out. The experimental results show that, compared with the original YOLOv5s algorithm, the average accuracy of the improved YOLO-M3C algorithm is improved by 0.123, and the recall rate is the same. These results verify that the algorithm YOLO-M3C has excellent performance in target detection and recognition, which can improve accuracy and confidence, while reducing false detection and missing detection, and effectively meet the needs of helmet-wearing detection.

engineering, electrical & electronic,physics, applied,computer science, information systems

What problem does this paper attempt to address?

### Problems Addressed by the Paper

This paper aims to address the issue of safety helmet detection in the field of power operations. Specifically, due to the weak safety awareness of workers and the failure to standardize the wearing of safety helmets, there are safety hazards. To effectively solve this problem, the paper proposes an improved lightweight neural network algorithm—YOLO-M3C.

#### Main Contributions:
1. **Replacing the Backbone Network**: Using MobileNetV3 to replace the backbone network of YOLOv5s for feature extraction, significantly reducing the network's computational load and making the network more lightweight.
2. **Introducing CA Attention Mechanism**: Introducing the CA (Coordinate Attention) mechanism in the backbone network to enhance the ability to focus on detection targets.
3. **Knowledge Distillation**: Applying knowledge distillation technology to improve the performance of the lightweight model, combining classification, confidence, and bounding box scales to transfer output feature knowledge.

Through these improvements, the YOLO-M3C algorithm not only improves detection accuracy and recall rate but also significantly reduces the model size and computational load, enabling deployment on low-computing-power platforms to meet the needs of practical application scenarios.