高级检索

基于改进YOLOv7的煤矿输送带上异物检测方法

Foreign object detection for coal mine conveyor belts based on improved YOLOv7

  • 摘要: 矿井输送带异物检测对于保障煤矿安全生产和提升自动化水平具有重要意义。然而,受井下光照不足、强点光源干扰以及背景复杂等因素影响,传统机器视觉方法在异物识别中存在检测精度低、漏检率高等问题,尤其在面对多尺度及小尺寸异物目标时效果不佳,同时有限的异物样本也制约了模型的训练效果。为此,提出一种融合特征增强结构与聚焦线性注意力机制的改进型YOLOv7异物检测算法。首先,在主干网络中引入构建的特征增强模块,通过多尺度并行结构与可切换空洞卷积扩展感受野,增强模型对小目标异物的特征提取能力;其次,融合聚焦线性注意力机制,有效抑制复杂背景干扰,提升模型对关键异物区域的关注力与判别能力;然后,设计高效动态蛇形卷积模块,以适应条状异物的几何特性,并提升特征融合阶段的语义表达能力;此外,采用具有方向感知能力的SIOU损失函数,增强边界框的拟合精度与训练收敛效率;最后,引入迁移学习策略,在通用数据集上进行预训练,并迁移至煤矿输送带异物检测任务中,以缓解小样本数据带来的训练难题。在CUMT-BELT数据集上的试验结果表明,本文提出的算法检测精度达到了89.3%,相较于基线模型YOLOv7提升了9.3%。同时,研究提出的算法在检测精度、召回率和mAP@0.5等指标上均显著优于其他主流的检测方法,尤其在多尺度、小目标及条状异物检测任务中表现出更强的鲁棒性与适应性,验证了所提方法在复杂矿井场景下的有效性与实用性。

     

    Abstract: Foreign object detection on mine conveyor belts is of significant importance for ensuring coal mine safety production and enhancing automation levels. However, traditional machine vision methods are limited by low detection accuracy, high missed detection rates, and inadequate performance in identifying multi-scale and small-sized foreign objects, due to challenges such as insufficient underground illumination, strong point light interference, and complex backgrounds. Additionally, the limited availability of foreign object samples constrains model training effectiveness. To address these issues, an improved YOLOv7 foreign object detection algorithm is proposed, which integrates a feature enhancement structure and a focused linear attention mechanism. First, a feature enhancement module with a multi-scale parallel architecture and switchable atrous convolution is introduced in the backbone network to expand the receptive field and improve small target feature extraction. Second, a focused linear attention mechanism is incorporated to suppress complex background interference while enhancing attention to critical foreign object regions. Third, an efficient dynamic serpentine convolution module is designed to adapt to the geometric characteristics of strip-shaped objects and strengthen semantic representation during feature fusion. Furthermore, a direction-aware SIOU loss function is employed to improve bounding box fitting accuracy and training convergence efficiency. Finally, a transfer learning strategy is implemented, where pre-training on generic datasets is leveraged before fine-tuning for coal mine conveyor belt detection tasks to mitigate small-sample training challenges. Experimental results on the CUMT-BELT dataset demonstrate that the proposed algorithm achieves a Precision of 89.3%, which represents a 9.3% improvement over the baseline YOLOv7 model. The method is also shown to significantly outperform other mainstream detection approaches in precision, recall, and mAP@0.5 metrics, while particularly demonstrating superior robustness and adaptability in multi-scale, small-target, and strip-shaped object detection tasks, thereby validating its effectiveness and practicality in complex mining scenarios.

     

/

返回文章
返回