关键词:
煤矸石分拣
目标检测
实时性
YOLOv8n
摘要:
为在煤矸石分拣领域实现检测性能与资源消耗的平衡,提出一种基于改进YOLOv8的适用于低性能检测平台的高效实时轻量化目标检测算法。首先以YOLOv8n为基础网络架构,引入MobileNetv3替换原有的主干网络,利用其轻量级结构特性降低模型参数量及运算量,提高模型检测速度;其次引入特征增强网络BIFPN模块,通过多尺度特征融合来弥补引入轻量级网络带来的检测精度损失,实现在保证检测精度的情况下完成模型轻量化;最后引入Inner-CIoU边界框回归损失函数平衡不同质量图像的训练结果,提高模型的定位能力,进一步提高检测精度及速度。为验证改进算法的有效性,进行了实验对比分析,将其与YOLOv3-tiny,YOLOv5n,YOLOv7以及YOLOv8n等算法在自建数据集上进行对比。实验结果表明,该算法展现出了最优的综合检测性能,在保证检测精度的前提下,其参数量降低到1188725,相较于YOLOv8n减少了60.46%,运算量由原模型的8.1 GFLOPs降低到2.8 GFLOPs,FPS由YOLOv8n的86.02 Hz提升到216.58 Hz。研究表明,该算法是一种高效实时轻量化煤矸石检测算法,综合检测性能有效提高,实现了模型检测性能与计算资源消耗的平衡,在煤矸石检测领域有较大的潜力和优越性。