基于改进YOLOv11n的轻量化刮板链条检测算法

Lightweight scraper chain detection algorithm based on improved YOLOv11n

  • 摘要: 针对现有基于深度学习的刮板输送机刮板链条检测方法在煤矿井下低照度条件下存在检测精度低、模型过于复杂、部署与维护难度高等问题,提出一种基于改进YOLOv11n的轻量化链条检测算法——YOLO−Chain。首先,构建图像边缘信息增强模块对YOLOv11n的C3k2模块进行优化,有效提取链条图像中的边缘特征。然后,采用加权双向特征金字塔网络(BiFPN)替换YOLOv11n的颈部网络,从而有效减少模型的参数量,降低模型复杂度。最后,引入轻量化检测头,捕捉井下复杂场景中链条尺度变化的细微特征,进一步减少冗余参数和模型复杂度,提高轻量化模型检测性能,为后续链条的故障检测提供支持。在山西某煤矿单一场景刮板链条图像数据集上的实验结果表明,相较于原始YOLOv11n模型,YOLO−Chain在mAP@0.5:0.95指标上提高了3.7%,参数量和运算量分别降低了35%和10%,模型大小减少了31%。与目前主流的YOLO系列,SSD,Faster RCNN,RT−DETR−R18等模型相比,YOLO−Chain在多项指标上均具优势。在来源于多家煤矿、具备低照度、烟雾干扰、粉尘遮蔽及部分遮挡等复杂工况下的多场景链条图像数据集上的实验结果表明,YOLO−Chain的F1和mAP@0.5分别较YOLOv11n模型提高了0.2%和0.5%,且运算速度提高8 帧/s,表现出良好的通用性和泛化能力。

     

    Abstract: To address the issues of low detection accuracy, excessive model complexity, and high deployment and maintenance difficulty in existing scraper conveyor chain detection methods based on deep learning under low illumination conditions in coal mines, a lightweight chain detection algorithm based on improved YOLOv11n—YOLO-Chain—was proposed. First, an image edge information enhancement module was constructed to optimize the C3k2 module of YOLOv11n, effectively extracting edge features from chain images. Then, a weighted Bidirectional Feature Pyramid Network (BiFPN) was used to replace the neck network of YOLOv11n, thereby effectively reducing the number of model parameters and lowering model complexity. Finally, a lightweight detection head was introduced to capture subtle features of chain scale variations in complex underground scenarios, further reducing redundant parameters and model complexity, improving the detection performance of the lightweight model, and providing support for subsequent chain fault detection. Experimental results on a single-scenario scraper chain image dataset from a coal mine in Shanxi showed that, compared with the original YOLOv11n model, YOLO-Chain improved the mAP@0.5:0.95 accuracy by 3.7%, while reducing the number of parameters and computational load by 35% and 10%, respectively, and decreasing the model size by 31%. Compared with current mainstream models such as the YOLO series, SSD, Faster RCNN, and RT-DETR-R18, YOLO-Chain also demonstrated advantages in multiple indicators. Experimental results on a multi-scenario chain image dataset collected from multiple coal mines under complex working conditions such as low illumination, smoke interference, dust occlusion, and partial obstruction showed that the F1-score and mAP@0.5 of YOLO-Chain increased by 0.2% and 0.5%, respectively, compared with YOLOv11n, with arithmetic speed increased by 8, demonstrating good applicability and generalization ability.

     

/

返回文章
返回