首页|基于MBI-YOLOv8的煤矸石目标检测算法研究

基于MBI-YOLOv8的煤矸石目标检测算法研究

扫码查看
为在煤矸石分拣领域实现检测性能与资源消耗的平衡,提出一种基于改进YOLOv8 的适用于低性能检测平台的高效实时轻量化目标检测算法.首先以YOLOv8n为基础网络架构,引入MobileNetv3 替换原有的主干网络,利用其轻量级结构特性降低模型参数量及运算量,提高模型检测速度;其次引入特征增强网络BIFPN模块,通过多尺度特征融合来弥补引入轻量级网络带来的检测精度损失,实现在保证检测精度的情况下完成模型轻量化;最后引入Inner-CIoU边界框回归损失函数平衡不同质量图像的训练结果,提高模型的定位能力,进一步提高检测精度及速度.为验证改进算法的有效性,进行了实验对比分析,将其与 YOLOv3-tiny,YOLOv5n,YOLOv7 以及YOLOv8n等算法在自建数据集上进行对比.实验结果表明,该算法展现出了最优的综合检测性能,在保证检测精度的前提下,其参数量降低到 1 188 725,相较于YOLOv8n减少了 60.46%,运算量由原模型的 8.1 GFLOPs降低到 2.8 GFLOPs,FPS由YOLOv8n的 86.02 Hz提升到 216.58 Hz.研究表明,该算法是一种高效实时轻量化煤矸石检测算法,综合检测性能有效提高,实现了模型检测性能与计算资源消耗的平衡,在煤矸石检测领域有较大的潜力和优越性.
Research on gangue target detection algorithm based on MBI-YOLOv8
To achieve a balance between detection performance and resource consumption in the gangue sorting domain,an efficient,real-time,lightweight object detection algorithm based on an improved YOLOv8 was proposed,suitable for low-performance detection platforms.This algorithm built on the YOLOv8n architecture and incorporated MobileNetv3 to replace the original backbone network,leveraging its lightweight structure to reduce model parameters and computational load,thereby enhancing detection speed.Additionally,the algorithm integrated the BIFPN module for feature enhancement,which employed multi-scale feature fusion to compensate for the loss of detection accuracy associated with the lightweight network,thus achieving model lightweighting while maintaining detection accuracy.Furthermore,the Inner-CIoU bounding box regression loss function was introduced to balance the training results of images with varying qualities,improving the model's localization capability and further enhancing detection accuracy and speed.To validate the effectiveness of the proposed algorithm,experiments were conducted to compare it with YOLOv3-tiny,YOLOv5n,YOLOv7,and YOLOv8n on a custom dataset.Experimental results demonstrated that the proposed algorithm exhibited optimal overall detection performance.While maintaining detection accuracy,the model's parameter count was reduced to 1,188,725,representing a 60.46%decrease compared to YOLOv8n.The computational load was reduced from 8.1 GFLOPs to 2.8 GFLOPs,and the FPS increased from 86.02 Hz to 216.58 Hz.This research indicated that the proposed algorithm is a highly efficient,real-time,lightweight gangue detection method with significant potential and advantages in balancing detection performance and computational resource consumption.

gangue sortingobject detectionreal-time performanceYOLOv8n

李珍峰、符世琛、徐乐、孟博、张昕、秦建军

展开 >

北京建筑大学机电与车辆工程学院,北京 102616

麦克马斯特大学工程实践与技术学院,汉密尔顿 L8S4L8

煤矸石分拣 目标检测 实时性 YOLOv8n

2024

图学学报
中国图学学会

图学学报

CSTPCD北大核心
影响因子:0.73
ISSN:2095-302X
年,卷(期):2024.45(6)