基于极致轻量化YOLOv8n的井下输送带异物检测方法

Foreign object detection method for underground conveyor belts based on an ultra-lightweight YOLOv8n

  • 摘要: 采用深度学习技术对输送带异物进行实时、精准检测,是保障带式输送机安全稳定运行的关键环节。常用的YOLO系列模型在轻量化与检测精度间难以平衡,其高计算复杂度与参数量难以很好地适配井下边缘计算设备的资源限制。针对该问题,以YOLOv8n为基础网络进行轻量化设计,构建了极致轻量化YOLOv8n——YOLOv8−PCAS。将YOLOv8n主干网络替换为PP−LCNet,实现主干轻量化;在C2f模块中引入连接结构优化的上下文锚点注意力(CAA)模块,增强对复杂异物形态的表征能力;引入平均池化下采样(ADown)策略,在有效压缩模型体积的同时,更好地保留关键语义信息;设计双检测头结构,去除冗余的大目标检测头,聚焦中小尺寸异物检测。基于煤矿井下异物数据CUMT−BelT和山西某煤矿井下监控视频对YOLOv8−PCAS进行训练和测试,实验结果表明:YOLOv8−PCAS的参数量为0.58×106个,为原始模型YOLOv8n的19.1%,运算量为3.6 GFLOPs,为YOLOv8n的44.4%,轻量化程度优于YOLOv7−tiny,YOLOv5n等主流模型及现有YOLOv8n轻量化改进方案;YOLOv8−PCAS能够有效检测出输送带上的锚杆、大块煤等目标,推理速度达357 帧/s,平均检测耗时2.8 ms,预测框与真实边界框交并比阈值为0.5 时的平均精度均值(mAP@0.5)为90.5%,满足工业现场对异物检测质量与时效的要求。

     

    Abstract: Real-time and accurate detection of foreign objects on conveyor belts using deep learning technology is crucial for ensuring the safe and stable operation of belt conveyors. Common YOLO series models struggle to balance lightweight design with detection accuracy, and their high computational complexity and parameter count hinder their deployment on resource-constrained underground edge computing devices. To address this problem, this study proposed an ultra-lightweight model, YOLOv8-PCAS, by applying a lightweight design to the YOLOv8n network. The backbone network of YOLOv8n was replaced with PP-LCNet to create a lightweight backbone. A Context Anchor Attention (CAA) module with an optimized connection structure was introduced into the C2f module to enhance the representation capability for complex shapes of foreign objects. The Average Pooling Down Sampling (ADown) strategy was incorporated to effectively reduce the model size while better preserving key semantic information. Furthermore, a dual detection head structure was designed, which removed the redundant large object detection head to focus on small and medium-sized foreign objects. The YOLOv8-PCAS model was trained and tested using the CUMT-BelT dataset of foreign objects from an underground coal mine and surveillance videos from a coal mine in Shanxi. The experimental results showed that the parameter count of YOLOv8-PCAS was approximately 0.58×106 (19.1% of the original YOLOv8n model), with a computational load of 3.6 GFLOPs (44.4% of YOLOv8n). Its lightweight performance surpassed that of mainstream models such as YOLOv7-tiny and YOLOv5n, as well as existing lightweight modifications of YOLOv8n. YOLOv8-PCAS effectively detected targets such as anchor bolts and lump coal on the conveyor belt, achieving an inference speed of 357 frames/s and an average detection time of 2.8 ms. The mean average precision reached 90.5% at an Intersection over Union (IoU) threshold of 0.5. The performance of YOLOv8-PCAS meets the industrial requirements for both detection quality and timeliness.

     

/

返回文章
返回