高级检索

基于深度学习的煤矿钻杆实时计数方法

Real time counting method for coal mine drill pipes based on deep learning

  • 摘要: 煤矿井下环境特殊,钻机工作场景复杂多变,存在强光、水汽、遮挡等因素干扰钻机识别效果,易出现误检漏检现象;同时钻机工作时钻机尾部常出现卡顿和无效运动等异常行为,影响钻杆计数准确性。针对上述问题,基于深度学习技术提出了一种煤矿钻杆实时计数方法,由基于AM-NT优化的钻机识别模型Drill-YOLOv8和基于两级判定区域的钻杆计数推理算法Pipe-Count 2部分组成。首先,从煤矿井下真实监控录像中采集钻机工作录像视频,标注并构建煤矿钻杆计数图像数据集CMDPC;然后,构建改进的钻机识别模型Drill-YOLOv8,引入Triplet注意力提升主干网络对钻机目标的特征提取能力,改进Slim-Neck网络结构平衡模型精度和复杂度,设计DDA Head模块替换检测头结构以提升模型头部对钻机目标多维度信息的关注度;最后,设计以两级判定区域统计钻机尾部有效运动次数为原理的钻杆计数推理方法Pipe-Count,依据Drill-YOLOv8检测器结果自动适应生成两级判定区域,使用ByteTrack实现钻机尾部目标追踪,通过判定区域更新钻机尾部运动状态,从而间接实现钻杆实时计数。在CMDPC数据集上的试验表明:改进的Drill-YOLOv8模型在mAP@0.5和mAP@0.5:0.95指标分别提升3.0%和2.7%,有效解决了强光、水汽和遮挡环境下钻机头部和钻杆目标误检漏检问题,检测速度为86帧/s;计数推理算法Pipe-Count的加权平均误计率为2%,面对多场景数据表现出良好的鲁棒性,且处理速度达到40帧/s,满足实时计数要求。

     

    Abstract: The underground environment of coal mines is special, and the working scene of drilling rigs is complex and variable. There are often factors such as strong light, water vapor, and obstruction that interfere with the recognition effect of drilling rigs, which can easily lead to false or missed detections; At the same time, abnormal behaviors such as lagging and ineffective movement often occur at the tail of the drilling rig during operation, which affects the accuracy of drill Pipe Counting. A real-time counting method for coal mine drill pipes based on deep learning technology is proposed to address the above issues. It consists of two parts: the drill recognition model Drill-YOLOv8 optimized based on AM-NT and the drill pipe counting inference algorithm Pipe Count based on two-level judgment regions. Firstly, collect the working video of the drilling rig from the real monitoring footage of the coal mine underground, annotate and construct the Coal Mine drill Pipe Counting image dataset CMDPC; Then, an improved drilling rig recognition model Drill-YOLOv8 is constructed, which enhances the feature extraction ability of the backbone network for drilling rig targets by introducing triplet attention. The Slim-Neck network structure is improved to balance the accuracy and complexity of the model, and the DDA Head module is designed to replace the detection head structure to enhance the attention of the model head to the multi-dimensional information of drilling rig targets; Finally, a pipe count inference method called Pipe Count is designed based on the principle of calculating the effective number of movements at the tail of the drilling rig using a two-level polygon judgment area. Based on the results of the Drill-YOLOv8 detector, a two-level judgment area is automatically generated, and ByteTrack method is used to track the target at the tail of the drilling rig. By updating the motion status of the tail of the drilling rig through the judgment area, real-time counting of the drilling rig is indirectly achieved. Experiments on the CMDPC dataset show that the improved Drill-YOLOv8 model performs well in mAP@0.5 The mAP @ 0.5:0.95 index has increased by 3.0% and 2.7% respectively, effectively solving the problem of false detection and missed detection of drilling head and drill pipe targets under strong light, water vapor, and occlusion environments, and the detection speed has reached 86 frames per second; At the same time, the weighted average error rate of the counting inference algorithm Pipe Count is 2%, showing good robustness against multi scene data, and the processing speed reaches 40 frames per second, meeting real-time counting requirements.

     

/

返回文章
返回