基于改进YOLO11的煤矿钻杆智能化计数算法

    Intelligent counting algorithm for coal mine drill pipes based on improved YOLO11

    • 摘要: 为解决煤矿井下钻杆拆卸过程中人工计数效率低、误差率高以及现有视觉计数方法在低光照、粉尘干扰等复杂环境下鲁棒性不足的问题,提出了一种基于改进YOLO11的煤矿钻杆智能检测与计数方法YOLO11−DP(YOLO11−Drill Pipe)。检测阶段,对YOLO11网络结构进行轻量化多尺度特征增强,引入轻量化EfficientNetV2主干网络,集成移动翻转瓶颈卷积(MBConv)和融合移动翻转瓶颈卷积(Fused−MBConv)模块,并采用分层注意力机制,在浅层引入高效通道注意力机制(ECA)强化边缘特征响应,中高层引入卷积注意力模块(CBAM)提升目标区域的聚焦能力以增强多尺度特征提取和细节捕捉能力。特征融合阶段,采用改进的PAN−FPN(Path Aggregation Network−Feature Pyramid Network)结构,使用深度可分离卷积(DWConv)和残差模块代替标准卷积,有效提升了模型对钻杆密集排列和遮挡场景的特征区分能力;设计了旋转框标注策略精确描述任意角度的钻杆目标,引入对角度变化敏感的KLD(Kullback−Leibler Divergence)损失函数提升旋转目标检测精度;采用K−means聚类算法分析数据集的高宽比分布,并优化锚框配置,使模型更契合钻杆细长的几何特征。计数阶段,结合DeepSORT目标跟踪算法,通过位移、角度、速度3个特征构建多条件联合判定机制,有效抑制由遮挡、轨迹中断、动态模糊及相似动作干扰导致的计数错误。现场试验结果表明,与原始YOLO11模型相比,YOLO11−DP模型对于倾斜的钻杆能够较为精准地覆盖,引入所有改进策略后模型在钻杆检测任务中平均精度均值mAP@0.5由76.8%提高至88.2%,计数误差率降低至6.4%。

       

      Abstract: To address the problems of low efficiency and high error rate of manual counting during drill pipe disassembly in underground coal mines, as well as the insufficient robustness of existing visual counting methods in complex environments such as low illumination and dust interference, an intelligent detection and counting method for coal mine drill pipes based on improved YOLO11, named YOLO11-DP (YOLO11-Drill Pipe), is proposed. In the detection stage, the lightweight multi-scale feature enhancement is carried out on the YOLO11 network structure. The lightweight EfficientNetV2 backbone is introduced, integrating the Mobile Inverted Bottleneck Convolution (MBConv) and Fused Mobile Inverted Bottleneck Convolution (Fused-MBConv) modules. A hierarchical attention mechanism is adopted: the efficient channel attention (ECA) is introduced in shallow layers to strengthen edge feature response, and the convolutional block attention module (CBAM) is applied in middle and high layers to enhance the focusing ability of target regions, so as to improve multi-scale feature extraction and detail capture capability. In the feature fusion stage, an improved path aggregation network-feature pyramid network (PAN-FPN) structure is used, in which depthwise separable convolution (DWConv) and residual modules replace standard convolution, effectively improving the model’s feature discrimination ability for densely arranged and occluded drill pipes. A rotated bounding box annotation strategy is designed to accurately describe drill pipe targets at arbitrary angles, and the angle-sensitive Kullback-Leibler Divergence (KLD) loss function is introduced to improve the precision of rotated object detection. K-means clustering is used to analyze the aspect ratio distribution of the dataset and optimize anchor box configuration, making the model more suitable for the slender geometric characteristics of drill pipes. In the counting stage, combined with the DeepSORT object tracking algorithm, a multi-condition joint decision mechanism is constructed based on three features: displacement, angle and velocity, which effectively suppresses counting errors caused by occlusion, trajectory interruption, motion blur and similar action interference. Field test results show that compared with the original YOLO11 model, the YOLO11-DP model can more accurately cover inclined drill pipes. After introducing all the improved strategies, the mAP@0.5 of the model in the drill pipe detection task is increased from 76.8% to 88.2%, and the counting error rate is reduced to 6.4%.

       

    /

    返回文章
    返回