基于改进YOLOv8的煤矿输送带异物检测OA北大核心CSTPCD
Foreign object detection of coal mine conveyor belt based on improved YOLOv8
现有基于深度学习的输送带异物检测模型较大,难以在边缘设备部署,且对不同尺寸异物和小目标异物存在错检、漏检情况.针对上述问题,提出一种基于改进YOLOv8的煤矿输送带异物检测方法.采用深度可分离卷积、压缩和激励(SE)网络将YOLOv8主干网络中C2f模块的Bottleneck重新构建为DSBlock,在保持模型轻量化的同时提升检测性能;为增强对不同尺寸目标物体信息的获取能力,引入高效通道注意力(ECA)机制,并对ECA的输入层进行自适应平均池化和自适应最大池化操作,得到跨通道交互MECA模块,以增强模块的全局视觉信息,进一步提升异物识别精度;将YOLOv8的3个检测头修改为4个轻量化小目标检测头,以增强对小目标的敏感性,有效降低小目标异物的漏检率和错检率.实验结果表明:改进YOLOv8的精确度达91.69%,mAP@50达92.27%,较YOLOv8分别提升了 3.09%和4.07%;改进YOLOv8的检测速度达73.92帧/s,可充分满足煤矿输送带异物实时检测的需求;改进YOLOv8的精确度、mAP@50、参数量、权重大小和每秒浮点运算数均优于 SSD,Faster-RCNN,YOLOv5,YOLOv7-tiny 等主流目标检测算法.
The existing deep learning based foreign object detection models for conveyor belts are relatively large and difficult to deploy on edge devices.There are errors and omissions in detecting foreign objects of different sizes and small objects.In order to solve the above problems,a foreign object detection method for coal mine conveyor belts based on improved YOLOv8 is proposed.The depthwise separable convolution,squeeze-and-excitation(SE)networks are used to reconstruct the Bottleneck of the C2f module in the YOLOv8 backbone network as a DSBlock,which improves the detection performance while keeping the model lightweight.To enhance the capability to obtain information from objects of different sizes,an efficient channel attention(ECA)mechanism is introduced.The input layer of ECA is subjected to adaptive average pooling and adaptive maximum pooling operations to obtain a cross channel interactive MECA module,which enhances the global visual information of the module and further improves the precision of foreign object recognition.The method modifies the 3 detection heads of YOLOv8 to 4 lightweight small object detection heads to enhance sensitivity to small objects and effectively reduce the missed and false detection rates of small object foreign objects.The experimental results show that the improved YOLOv8 achieves a precision of 91.69%,mAP@50 reached 92.27%,an increase of 3.09%and 4.07%respectively compared to YOLOv8.The detection speed of improved YOLOv8 reaches 73.92 frames/s,which can fully meet the demand for real-time detection of foreign objects on conveyor belts in coal mines.The improved YOLOv8 outperforms mainstream object detection algorithms such as SSD,Faster-RCNN,YOLOv5,and YOLOv7-tiny in terms of precision,mAP@50,number of parameters,weight size,and number of floating point operations.
洪炎;汪磊;苏静明;汪瀚涛;李木石
安徽理工大学电气与信息工程学院,安徽淮南 232001
矿山工程
输送带异物检测YOLOv8SE网络高效通道注意力机制轻量化小目标检测自适应平均池化自适应最大池化
foreign object detection on conveyor beltsYOLOv8SE networkefficient channel attention mechanismlightweightsmall object detectionadaptive average poolingadaptive maximum pooling
《工矿自动化》 2024 (006)
61-69 / 9
国家重点研发计划项目(2021YFD2000204);国家自然科学基金项目(12304236,32301688,52174141);安徽数字农业工程技术研究中心开放项目(AHSZNYGCZXKF021);大学生创新创业基金项目(202210361053,202310361037);安徽理工大学研究生创新基金项目(2024cx2067).
评论