Yolov8n-FADS: A Study for Enhancing Miners' Helmet Detection Accuracy in Complex Underground Environments

Zhibo Fu,Jierui Ling,Xinpeng Yuan,Hao Li,Hongjuan Li,Yuanfei Li

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

IF: 3.9

2024-06-11

Sensors

Abstract:A new algorithm, Yolov8n-FADS, has been proposed with the aim of improving the accuracy of miners' helmet detection algorithms in complex underground environments. By replacing the head part with Attentional Sequence Fusion (ASF) and introducing the P2 detection layer, the ASF-P2 structure is able to comprehensively extract the global and local feature information of the image, and the improvement in the backbone part is able to capture the spatially sparsely distributed features more efficiently, which improves the model's ability to perceive complex patterns. The improved detection head, SEAMHead by the SEAM module, can handle occlusion more effectively. The Focal Loss module can improve the model's ability to detect rare target categories by adjusting the weights of positive and negative samples. This study shows that compared with the original model, the improved model has 29% memory compression, a 36.7% reduction in the amount of parameters, and a 4.9% improvement in the detection accuracy, which can effectively improve the detection accuracy of underground helmet wearers, reduce the workload of underground video surveillance personnel, and improve the monitoring efficiency.

engineering, electrical & electronic,chemistry, analytical,instruments & instrumentation

What problem does this paper attempt to address?

The main objective of this paper is to improve the accuracy of miner helmet detection in complex underground environments. To achieve this goal, the research team proposed an improved algorithm model called Yolov8n-FADS (Yolov8n with Fusion Attention Detection System). This model optimizes the shortcomings of existing technologies in helmet detection under challenges such as low light, cluttered environments, and varying monitoring distances.

Specifically, the main contributions of Yolov8n-FADS include:

1. **Model Structure Improvement**:
- Introduced Dilated Reparam Block and RepNCSPELAN modules in the backbone network to enhance the model's ability to capture sparsely distributed features and improve convergence speed.
- Adopted the ASF (Attentional Scale Sequence Fusion) structure and P2 detection layer in the head network to better extract global and local feature information and handle multi-scale detection tasks.
- Used the SEAMHead module to address occlusion issues by reducing background interference through the introduction of a multi-head attention mechanism.

2. **Loss Function Optimization**:
- Introduced the Focal Loss module to adjust the weights of positive and negative samples, improving the detection capability for rare category targets.
- Used the Focaler IoU loss function to improve the accuracy of bounding box regression by linearly mapping samples of different difficulty levels, allowing the model to focus on optimizing specific types of samples.

3. **Attention Mechanism**:
- Integrated the Triplet Attention mechanism to enhance feature fusion capability by processing channel attention, spatial attention, and cross-dimensional interaction through three branches, thereby improving overall performance.

In summary, through these improvements, the Yolov8n-FADS model can significantly reduce memory usage and parameter count while maintaining high detection accuracy. This effectively improves the detection accuracy of helmet wearers in underground environments, reduces the workload of video surveillance personnel, and enhances monitoring efficiency.