Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrack
In order to solve the problems of low accuracy and poor real-time performance of existing target tracking algorithms in the complex environment of coal mines, a YOLO-FasterNet+ByteTrack coal mine personnel tracking algorithm was proposed based on the Tracking by Detection (TBD) paradigm. Firstly, th...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | zho |
| Published: |
Editorial Office of Safety in Coal Mines
2025-01-01
|
| Series: | Meikuang Anquan |
| Subjects: | |
| Online Access: | https://www.mkaqzz.com/cn/article/doi/10.13347/j.cnki.mkaq.20240314 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841527931733540864 |
|---|---|
| author | Pengcheng QU Jingzhao LI Zechao LIU |
| author_facet | Pengcheng QU Jingzhao LI Zechao LIU |
| author_sort | Pengcheng QU |
| collection | DOAJ |
| description | In order to solve the problems of low accuracy and poor real-time performance of existing target tracking algorithms in the complex environment of coal mines, a YOLO-FasterNet+ByteTrack coal mine personnel tracking algorithm was proposed based on the Tracking by Detection (TBD) paradigm. Firstly, the FasterNet-Block feature extraction module was constructed to improve the Backbone of YOLOv7 and improve the real-time performance of the object detection stage. Then, the CBAM attention mechanism was introduced into Neck to improve the feature perception ability of the model in complex scenes. Then, Soft-NMS is introduced in the decoding stage of object detection to optimize the detection accuracy of the model in personnel overlapping scenario. Finally, in the target tracking stage, a multi-target motion feature prediction mechanism fused with GRU and Kalman filter was designed to solve the problem of target ID flipping caused by personnel overlap and occlusion, which effectively improved the accuracy of coal mine personnel tracking. Experimental results show that the average accuracy of YOLO-FasterNet is increased by 3.6% and the detection speed is increased by 8.2FPS compared with YOLOv7 on the coal mine personnel dataset, and the MOTA value of the proposed target tracking algorithm is increased by 1.7% and the IDSW is reduced by 149 times compared with ByteTrack on the custom tracking dataset GBMOT. |
| format | Article |
| id | doaj-art-d3dd19e53c3e47cc866e6b6126e7aca8 |
| institution | Kabale University |
| issn | 1003-496X |
| language | zho |
| publishDate | 2025-01-01 |
| publisher | Editorial Office of Safety in Coal Mines |
| record_format | Article |
| series | Meikuang Anquan |
| spelling | doaj-art-d3dd19e53c3e47cc866e6b6126e7aca82025-01-15T04:32:08ZzhoEditorial Office of Safety in Coal MinesMeikuang Anquan1003-496X2025-01-0156119520510.13347/j.cnki.mkaq.20240314lMKAQ20240314Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrackPengcheng QU0Jingzhao LI1Zechao LIU2School of Artificial Intelligence, Anhui University of Science and Technology, Huainan 232001, ChinaSchool of Artificial Intelligence, Anhui University of Science and Technology, Huainan 232001, ChinaSchool of Electrical and Information Engineering, Anhui University of Science and Technology, Huainan 232001, ChinaIn order to solve the problems of low accuracy and poor real-time performance of existing target tracking algorithms in the complex environment of coal mines, a YOLO-FasterNet+ByteTrack coal mine personnel tracking algorithm was proposed based on the Tracking by Detection (TBD) paradigm. Firstly, the FasterNet-Block feature extraction module was constructed to improve the Backbone of YOLOv7 and improve the real-time performance of the object detection stage. Then, the CBAM attention mechanism was introduced into Neck to improve the feature perception ability of the model in complex scenes. Then, Soft-NMS is introduced in the decoding stage of object detection to optimize the detection accuracy of the model in personnel overlapping scenario. Finally, in the target tracking stage, a multi-target motion feature prediction mechanism fused with GRU and Kalman filter was designed to solve the problem of target ID flipping caused by personnel overlap and occlusion, which effectively improved the accuracy of coal mine personnel tracking. Experimental results show that the average accuracy of YOLO-FasterNet is increased by 3.6% and the detection speed is increased by 8.2FPS compared with YOLOv7 on the coal mine personnel dataset, and the MOTA value of the proposed target tracking algorithm is increased by 1.7% and the IDSW is reduced by 149 times compared with ByteTrack on the custom tracking dataset GBMOT.https://www.mkaqzz.com/cn/article/doi/10.13347/j.cnki.mkaq.20240314coal mine personnel positioning systemmultiple object trackingyolov7attention mechanismgated recurrent unit |
| spellingShingle | Pengcheng QU Jingzhao LI Zechao LIU Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrack Meikuang Anquan coal mine personnel positioning system multiple object tracking yolov7 attention mechanism gated recurrent unit |
| title | Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrack |
| title_full | Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrack |
| title_fullStr | Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrack |
| title_full_unstemmed | Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrack |
| title_short | Multi-target personnel tracking algorithm for coal mine based on improved YOLOv7 and ByteTrack |
| title_sort | multi target personnel tracking algorithm for coal mine based on improved yolov7 and bytetrack |
| topic | coal mine personnel positioning system multiple object tracking yolov7 attention mechanism gated recurrent unit |
| url | https://www.mkaqzz.com/cn/article/doi/10.13347/j.cnki.mkaq.20240314 |
| work_keys_str_mv | AT pengchengqu multitargetpersonneltrackingalgorithmforcoalminebasedonimprovedyolov7andbytetrack AT jingzhaoli multitargetpersonneltrackingalgorithmforcoalminebasedonimprovedyolov7andbytetrack AT zechaoliu multitargetpersonneltrackingalgorithmforcoalminebasedonimprovedyolov7andbytetrack |