宁夏工程技术2024，Vol.23Issue(2)：174-179,6.
基于改进YOLOv5s的煤矿输送带异物检测算法
An Improved YOLOv5s-based Foreign Object Detection Algorithm for Coal Mine Conveyor Belts
摘要
Abstract
In order to address the issues of large parameter size,high demand for computer resources,and limited types of detected foreign objects in existing detection models for coal mine conveyor belts,YOLOv5s object detection algorithm was optimized.Firstly,a lightweight convolutional neural network,ShuffleNetv2,was adopted as the backbone network of YOLOv5s to extract features from foreign object images,reducing the parameters of the model and improving network parallelism.Secondly,a bidirectional feature pyramid network was used as the feature fusion network to integrate detailed information from different feature map scales.Finally,a coordinate attention mechanism was added to enhance the feature extraction capability,strengthen the focus on foreign object targets,and thereby improve the detection accuracy of network model.Experimental results show that compared to the original model,the improved YOLOv5s-based object detection network model achieved a parameter compression of 3.6×106,and an 8.4%increase in detection frame rate,demonstrating that this method can achieve rapid and accurate foreign object detection on coal mine conveyor belts with fewer computational resources.关键词
煤矿输送带/异物检测/YOLOv5s/特征融合网络/注意力机制Key words
coal mine conveyor belt/foreign object detection/YOLOv5s/feature fusion network/attention mechanism分类
矿业与冶金引用本文复制引用
张炳建,郝洪涛,刘秀娟,李泽旭..基于改进YOLOv5s的煤矿输送带异物检测算法[J].宁夏工程技术,2024,23(2):174-179,6.基金项目
宁夏回族自治区重点研发计划项目(2023ZDYF0142) （2023ZDYF0142）
宁夏自然科学基金项目(2021AAC03046) （2021AAC03046）
