基于深度学习的矿井巷道人员计数技术
Mine roadway personnel counting technology based on deep learning
-
摘要: 在煤矿发生安全事故时需要明确掌握各个区域的人员情况，合理安排营救计划。使用YOLOv5作为目标检测器，结合改进的DeepSORT跟踪算法进行矿井人员跟踪，实现煤矿矿井各个巷道区域的人员计数；首先使用轻量化的全尺度特征学习Re-ID特征提取模型OSNet对DeepSORT进行优化，替换原有的CNN特征提取模块；然后采用检测器和OSNet特征提取模型单独训练的策略，实现了矿井复杂环境下稳定的跟踪效果；在此基础上，通过在视频画面中设置ROI区域和基准线来判断人员进出的情况，从而实现计数功能。为了有效训练和评估模型的性能，采集了10 000张煤矿矿井下各个巷道不同区域的图片用于训练和测试，改进后模型的MOTA为66.7%，优于改进前的63.4%；改进后速度为28.1 FPS，优于改进前的25.3 FPS。试验结果表明：改进后的模型可以有效地实现矿井人员计数，可以用到实际的生产环境中。Abstract: When a safety accident occurs in a coal mine, it is necessary to clearly understand the personnel situation in each area and reasonably arrange the rescue plan. In this paper, YOLOv5 is used as the target detector, combined with the improved DeepSORT tracking algorithm to track the mine personnel, and the personnel count in each roadway area of the mine is realized. Firstly, the lightweight full-scale feature learning Re-ID feature extraction model OSNet is used to optimize DeepSORT and replace the original CNN feature extraction module. Then, the strategy of training the detector and OSNet feature extraction model separately is adopted to achieve a stable tracking effect in the complex environment of the mine. On this basis, the ROI area and baseline are set in the video screen to judge the situation of people entering and leaving, so as to realize the counting function. In order to effectively train and evaluate the performance of the model, 10 000 pictures of different areas of various roadways under coal mines were collected for training and testing. The MOTA of the improved model was 66.7%, better than that of the former 63.4%. The improved speed is 28.1 FPS, which is better than the 25.3 FPS before the improvement. The experimental results show that the improved model can effectively count mine personnel and can be used in the actual production environment.
-
Keywords:
- mine personnel tracking /
- people counting /
- YOLOv5 /
- DeepSORT /
- OSNet
-
-
[1] 叶伟.煤矿井下目标定位的研究现状与展望[J].中国矿业，2021，30（1）：82-89. YE Wei. Status and prospects of research on positioning of targets in underground coal mines[J]. China Mining Magazine, 2021, 30（1）: 82-89.
[2] 陈代伟.基于UWB和RFID多源融合的煤矿人员精准定位系统设计[J].现代矿业，2021，37（12）：197-200. CHEN Daiwei. Design of coal mine personnel precise positioning system based on UWB and RFID multi-source fusion[J]. Modern Mining, 2021, 37（12）: 197-200.
[3] 徐桂涛.煤矿视频监控系统中人员目标跟踪算法的研究与实现[D].徐州：中国矿业大学，2019. [4] 姚娟.多目标跟踪算法及其在车辆计数中的应用研究[D].武汉：华中科技大学，2020. [5] 文奴，郭仁忠，贺彪.基于DCN-Mobile-YOLO模型的多车道车辆计数[J].深圳大学学报（理工版），2021， 38（6）：628-635. WEN Nu, GUO Renzhong, HE Biao. Multi-lane vehicle counting based on DCN-Mobile-YOLO model[J]. Journal of Shenzhen University（Science and Engineering）, 2021, 38（6）: 628-635.
[6] 李永上，马荣贵，张美月.改进YOLOv5s+DeepSORT的监控视频车流量统计[J].计算机工程与应用，2022， 58（5）：271-279. LI Yongshang, MA Ronggui, ZHANG Meiyue. Traffic monitoring video vehicle volume statistics method based on improved YOLOv5s+DeepSORT[J]. Computer Engineering and Applications, 2022, 58（5）: 271-279.
[7] 张译文，李琦.基于深度学习的羊只计数系统研究[J].内蒙古科技大学学报，2021，40（4）：370-374. ZHANG Yiwen, LI Qi. Research on sheep counting system based on embedded AI[J]. Journal of Inner Mongolia University of Science and Technology, 2021, 40（4）: 370-374.
[8] 何宇，李丹.基于YOLOv5和DeepSort的实时地铁客流量检测[J].电子测试，2022，36（2）：46-48. HE Yu, LI Dan. Real-time subway ridership detection based on YOLOv5 and DeepSort[J]. Electronic Test, 2022, 36（2）: 46-48.
[9] BEWLEY A, GE Z, OTT L, et al. Simple Online and Realtime Tracking[C]// 2016 IEEE International Conference on Image Processing（ICIP）. IEEE, 2016: 3464-3468. [10] WOJKE N, BEWLEY A, PAULUS D. Simple Online and Realtime Tracking with a Deep Association Metric[C]//2017 IEEE International Conference on Image Processing（ICIP）. IEEE, 2017: 3645-3649. -
期刊类型引用(3)
1. 瞿鹏程，李敬兆，刘泽朝. 基于改进YOLOv7和Byte Track的煤矿多目标人员跟踪算法. 煤矿安全. 2025(01): 195-205 . 
本站查看
2. 路洋，董立红，叶鸥. 基于自适应链接优化的井下行人抗遮挡跟踪方法研究. 工矿自动化. 2025(02): 65-75+137 . 百度学术
3. 王端，刘世平，王利军. 基于机器视觉的煤矿巷道人员定位研究. 矿山机械. 2024(01): 56-60 . 百度学术
其他类型引用(0)
计量
- 文章访问数: 32
- HTML全文浏览量: 1
- PDF下载量: 21
- 被引次数: 3
下载:
