A lightweight personnel detection method for underground coal mines
The underground environment of coal mines is complex and has more safety hazards. Personnel detection is an important part of ensuring safe production in coal mines and building smart mines. Commonly used detection algorithms have large parameter counts, high requirements on equipment arithmetic, an...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | zho |
| Published: |
Editorial Department of Coal Science and Technology
2025-04-01
|
| Series: | Meitan kexue jishu |
| Subjects: | |
| Online Access: | http://www.mtkxjs.com.cn/article/doi/10.12438/cst.2023-1617 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850128650125443072 |
|---|---|
| author | Shuai WANG Wei YANG Yuxiang LI Jiaqi WU Wei YANG |
| author_facet | Shuai WANG Wei YANG Yuxiang LI Jiaqi WU Wei YANG |
| author_sort | Shuai WANG |
| collection | DOAJ |
| description | The underground environment of coal mines is complex and has more safety hazards. Personnel detection is an important part of ensuring safe production in coal mines and building smart mines. Commonly used detection algorithms have large parameter counts, high requirements on equipment arithmetic, and are not satisfactory for application in low illumination environments in coal mines. To address the above problems, a lightweight personnel detection method YOLOv5-CWG is proposed for underground coal mine based on YOLOv5.Firstly, the coordinate attention mechanism (Coordinate Attention) embedded in the backbone network adaptively adjusts the weights of each channel in the feature map to enhance the expression ability of the features. It improves the attention to the personnel target to be detected under the unfavorable conditions of low illumination, serious dust influence and low contrast, and locates and identifies the personnel target more accurately. Secondly, the weighted multiscale feature fusion module (Weighted multiscale feature fusion moule) introduces learnable weights to give different attention to the feature layer. This enables the network to effectively fuse shallow positional features and high-level semantic information to enhance the information extraction capability and better distinguish between target and interference, improving the anti-interference capability of the model. Add a P2 layer detection head to improve the detection and localization accuracy of smaller targets. Introduce SIoU instead of the original loss function to accelerate model convergence. Finally, the introduction of the Ghost module to optimize the backbone network can reduce the computational and parametric quantities of the model without losing the model performance, improve the detection speed, and make the model easier to be deployed on resource-constrained devices. Experimental results show that the proposed YOLOv5-CWG algorithm achieves 97.5% mAP on the Underground Mine Personnel Detection Dataset (UMPDD). Compared with YOLOv5s, the mAP is improved by 7.3%, the computation amount is reduced by 27.6%, and the FPS is improved by 6.3. The proposed algorithm significantly improves the accuracy of personnel detection in underground coal mines, and efficiently solves the problem of difficult personnel detection caused by low brightness and uneven illumination. |
| format | Article |
| id | doaj-art-e76fc667053745c6936ac3f6b82867fc |
| institution | OA Journals |
| issn | 0253-2336 |
| language | zho |
| publishDate | 2025-04-01 |
| publisher | Editorial Department of Coal Science and Technology |
| record_format | Article |
| series | Meitan kexue jishu |
| spelling | doaj-art-e76fc667053745c6936ac3f6b82867fc2025-08-20T02:33:13ZzhoEditorial Department of Coal Science and TechnologyMeitan kexue jishu0253-23362025-04-0153438339210.12438/cst.2023-16172023-1617A lightweight personnel detection method for underground coal minesShuai WANG0Wei YANG1Yuxiang LI2Jiaqi WU3Wei YANG4School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083School of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083Hunan Institute of Advanced Technology, Changsha 410205, ChinaSchool of Artificial Intelligence, China University of Mining and Technology-Beijing, Beijing 100083School of Electronic and Information Engineering, Beijing Jiatong University, Beijing 100044, ChinaThe underground environment of coal mines is complex and has more safety hazards. Personnel detection is an important part of ensuring safe production in coal mines and building smart mines. Commonly used detection algorithms have large parameter counts, high requirements on equipment arithmetic, and are not satisfactory for application in low illumination environments in coal mines. To address the above problems, a lightweight personnel detection method YOLOv5-CWG is proposed for underground coal mine based on YOLOv5.Firstly, the coordinate attention mechanism (Coordinate Attention) embedded in the backbone network adaptively adjusts the weights of each channel in the feature map to enhance the expression ability of the features. It improves the attention to the personnel target to be detected under the unfavorable conditions of low illumination, serious dust influence and low contrast, and locates and identifies the personnel target more accurately. Secondly, the weighted multiscale feature fusion module (Weighted multiscale feature fusion moule) introduces learnable weights to give different attention to the feature layer. This enables the network to effectively fuse shallow positional features and high-level semantic information to enhance the information extraction capability and better distinguish between target and interference, improving the anti-interference capability of the model. Add a P2 layer detection head to improve the detection and localization accuracy of smaller targets. Introduce SIoU instead of the original loss function to accelerate model convergence. Finally, the introduction of the Ghost module to optimize the backbone network can reduce the computational and parametric quantities of the model without losing the model performance, improve the detection speed, and make the model easier to be deployed on resource-constrained devices. Experimental results show that the proposed YOLOv5-CWG algorithm achieves 97.5% mAP on the Underground Mine Personnel Detection Dataset (UMPDD). Compared with YOLOv5s, the mAP is improved by 7.3%, the computation amount is reduced by 27.6%, and the FPS is improved by 6.3. The proposed algorithm significantly improves the accuracy of personnel detection in underground coal mines, and efficiently solves the problem of difficult personnel detection caused by low brightness and uneven illumination.http://www.mtkxjs.com.cn/article/doi/10.12438/cst.2023-1617object detectionyolov5attention mechanismlightweightfeature fusion |
| spellingShingle | Shuai WANG Wei YANG Yuxiang LI Jiaqi WU Wei YANG A lightweight personnel detection method for underground coal mines Meitan kexue jishu object detection yolov5 attention mechanism lightweight feature fusion |
| title | A lightweight personnel detection method for underground coal mines |
| title_full | A lightweight personnel detection method for underground coal mines |
| title_fullStr | A lightweight personnel detection method for underground coal mines |
| title_full_unstemmed | A lightweight personnel detection method for underground coal mines |
| title_short | A lightweight personnel detection method for underground coal mines |
| title_sort | lightweight personnel detection method for underground coal mines |
| topic | object detection yolov5 attention mechanism lightweight feature fusion |
| url | http://www.mtkxjs.com.cn/article/doi/10.12438/cst.2023-1617 |
| work_keys_str_mv | AT shuaiwang alightweightpersonneldetectionmethodforundergroundcoalmines AT weiyang alightweightpersonneldetectionmethodforundergroundcoalmines AT yuxiangli alightweightpersonneldetectionmethodforundergroundcoalmines AT jiaqiwu alightweightpersonneldetectionmethodforundergroundcoalmines AT weiyang alightweightpersonneldetectionmethodforundergroundcoalmines AT shuaiwang lightweightpersonneldetectionmethodforundergroundcoalmines AT weiyang lightweightpersonneldetectionmethodforundergroundcoalmines AT yuxiangli lightweightpersonneldetectionmethodforundergroundcoalmines AT jiaqiwu lightweightpersonneldetectionmethodforundergroundcoalmines AT weiyang lightweightpersonneldetectionmethodforundergroundcoalmines |