工矿自动化2024，Vol.50Issue(3)：42-47,141,7.DOI:10.13272/j.issn.1671-251x.2024030009
基于MES-YOLOv5s的综采工作面大块煤检测算法
Large block coal detection algorithm for fully mechanized working face based on MES-YOLOv5s
摘要
Abstract
The objects in the fully mechanized working face have the features of high-speed motion,multi-scale,occlusion,etc.The existing object detection algorithms have problems such as low precision,large memory of models,and strong hardware dependence.In order to solve the above problems,a large block coal detection algorithm based on MES-YOLOv5s is proposed in fully mechanized working face.The method adopts a lightweight design,uses MobileNetV3 as the backbone network to reduce the memory occupied by the model and improve the detection speed on the CPU side.The method adds an efficient multi-scale attention(EMA)module to the neck network,fuses contextual information of different scales,and further reduces computational overhead.The method uses SIoU loss function instead of CIoU loss function to improve training speed and inference accuracy.The ablation experiment results show that MobileNetV3 significantly reduces the memory and detection time occupied by the model,but the mAP loss is severe.The EMA module and SIoU loss function can restore the precision of the loss to a certain extent,while ensuring that the model has a high detection speed on the CPU,meeting the real-time detection needs of coal mine underground objects.The comparative experimental results show that compared with DETR,YOLOv5n,YOLOv5s,and YOLOv7 models,the MES-YOLOv5s model has the best overall performance,with an mAP of 84.6％.The model occupies 11.2 MiB of memory and has a detection time of 31.8 ms on the CPU side.It can maintain high recall and precision in high-speed motion,multi-scale,occlusion,and multi-object working environments.关键词
综采工作面/目标检测/大块煤检测/YOLOv5s/MobileNetV3/高效多尺度注意力模块/SIoU损失函数Key words
fully mechanized working face/object detection/large block coal detection/YOLOv5s/MobileNetV3/efficient multi-scale attention module/SIoU loss function分类
矿业与冶金引用本文复制引用
徐慈强,贾运红,田原..基于MES-YOLOv5s的综采工作面大块煤检测算法[J].工矿自动化,2024,50(3):42-47,141,7.基金项目
国家重点研发计划资助项目(2020YFB1314003) （2020YFB1314003）
山西省自然科学基金资助项目(201801D121189). （201801D121189）
