基于改进YOLOv8s的矿用输送带异物检测方法

Improved YOLOv8s-based foreign object detection method for mine conveyor belts

  • 摘要: 针对矿井低照度环境下输送带异物检测算法存在的图像全局特征提取不足、模型参数量过大等问题,提出一种基于改进YOLOv8s的矿用输送带异物检测方法。利用VMamba和MobileNetv4对YOLOv8s进行改进:采用MobileNetv4改进主干网络,集成通用逆瓶颈(UIB)模块,通过高效倒置残差结构降低模型整体参数量,通过动态特征适应机制增强小目标场景的特征鲁棒性;通过VMamba的视觉状态空间(VSS)模块改进核心特征提取与融合模块C2f,通过状态空间模型和四向扫描机制高效捕捉图像中的全局上下文信息,增强模型对图像全局结构的理解;设计了参数共享轻量化检测头,使用分组归一化(GN)作为归一化卷积基本块,弥补模型轻量化所带来的精度损失。实验结果表明:改进YOLOv8s模型在自建数据集上的mAP@0.5达0.921,mAP@0.5:0.95达0.601,参数量较YOLOv8s减少27.7%,性能优于主流目标检测模型YOLOv11s,YOLOv10s等,可以满足矿用输送带异物检测需求。

     

    Abstract: In low-illumination mine environments, conveyor belt foreign object detection algorithms suffer from insufficient extraction of global image features and an excessive number of model parameters. A method for detecting foreign objects on mine conveyor belts based on an improved version of YOLOv8s was proposed. YOLOv8s was improved using VMamba and MobileNetv4: MobileNetv4 was employed to enhance the backbone network by integrating the Universal Inverted Bottleneck (UIB) module. The efficient inverted residual structure reduced the overall number of model parameters, and a dynamic feature adaptation mechanism was used to strengthen feature robustness in small-object scenarios. The core feature extraction and fusion module C2f was improved by VMamba's Visual State Space (VSS) module, which efficiently captured global contextual information in images through a state space model and four-directional scanning mechanism, enhancing the model’s understanding of global image structure. A parameter-sharing lightweight detection head was designed, using Group Normalization (GN) as the basic convolutional normalization block to compensate for accuracy loss caused by model lightweighting. Experimental results showed that the improved YOLOv8s model achieved an mAP@0.5 of 0.921 and an mAP@0.5:0.95 of 0.601 on a self-built dataset, reduced the number of parameters by 27.7% compared to original YOLOv8s, outperformed mainstream object detection models such as YOLOv11s and YOLOv10s, and met the requirements for foreign object detection on mine conveyor belts.

     

/

返回文章
返回