基于YOLOv11_OBB的煤矿钻杆计数方法

Method for counting coal mine drill pipes based on YOLOv11_OBB

  • 摘要: 针对煤矿井下打钻图像识别泛化能力差、钻杆计数不准确等问题,采集并标注了煤矿井下钻杆计数数据集CMDPC_OBB,提出了一种基于YOLOv11_OBB的煤矿钻杆计数方法。该方法包括打钻图像识别模型YOLOv11_OBB和场景自适应的钻杆计数算法2个部分。YOLOv11_OBB采用旋转边界框,精准捕获具有倾斜角度的打钻图像,通过L2正则化处理改进YOLOv11颈部网络,降低权重波动对特征融合的干扰,使模型训练更加稳定;场景自适应的钻杆计数算法通过追踪目标钻杆与钻机尾部关键点之间的运动轨迹及多条件判断峰值实现自动计数,减少了与计数无关的钻杆对计数准确率的影响;将YOLOv11_OBB学习到的图像特征作为潜在知识指导钻杆计数的推理逻辑。在CMDPC_OBB数据集上的实验结果表明,YOLOv11_OBB的图像识别精度为93.5%,与YOLOv5L_OBB,YOLOv8L_OBB相比分别提升了8.2%,3.0%;钻杆计数算法的准确率为97.96%,模型识别速度为34帧/s,计数速度为79帧/s,满足实时计数要求。

     

    Abstract: To address the issues of poor generalization in downhole drilling image recognition and inaccurate drill pipe counting in coal mines, this study collected and annotated a dedicated dataset CMDPC_OBB, and proposed a method for drill pipe counting based on YOLOv11_OBB. The method consisted of two components: the YOLOv11_OBB-based drilling image recognition model and a scene-adaptive drill pipe counting algorithm. YOLOv11_OBB used rotated bounding boxes to accurately capture drilling images with inclined angles. By applying L2 regularization to improve the neck network of YOLOv11, weight fluctuation interference in feature fusion was reduced, ensuring more stable model training. The scene-adaptive counting algorithm tracked the motion trajectory between target drill pipes and key points at the drill rig's tail while employing multi-condition peak judgment to achieve automatic counting, thereby minimizing the impact of irrelevant pipes on accuracy. In addition, the image features learned by YOLOv11_OBB served as latent knowledge to guide the counting logic. Experimental results on the CMDPC_OBB dataset show that YOLOv11_OBB achieves an image recognition accuracy of 93.5%, outperforming YOLOv5L_OBB and YOLOv8L_OBB by 8.2% and 3.0%, respectively. The counting algorithm achieves an accuracy of 97.96%, with a model recognition speed of 34 FPS and a counting speed of 79 FPS, meeting real-time requirements.

     

/

返回文章
返回