基于改进YOLOv8n和ByteTrack的电子元器件分拣多目标跟踪算法研究

Multi-object Tracking Algorithm for Electronic Component Sorting Based on Improved YOLOv8n and ByteTrack

  • 摘要: 针对3C电子元器件分拣场景中普遍存在的小目标、密集排列等问题对多目标跟踪效果产生的影响, 提出一种基于改进YOLOv8n和ByteTrack的轻量化多目标跟踪算法。在目标检测环节, 针对小目标检测难、目标之间相互遮挡等问题, 以YOLOv8n模型为基础, 采用Gold-YOLO模块替换原网络中的Neck模块以强化模型多尺度特征融合能力, 并在Gold-YOLO的N3输出层嵌入ACmix注意力机制模块, 增强对小目标及遮挡目标的特征提取能力;在目标跟踪环节, 以ByteTrack模型为基础, 优化算法参数并增加嵌套框过滤算法, 以有效抑制重叠检测框导致的误匹配, 从而提升目标跟踪鲁棒性。利用自建数据集进行改进模型的性能验证, 结果表明, 改进后的YOLOv8n与ByteTrack目标跟踪算法在电子元器件分拣场景中应用性能优异。在自建数据集中, 目标检测的平均精度均值mAP@50达91.32%, 多目标跟踪准确率(MOTA)达86.72%, 身份识别F1分数(IDF1)达93.13%, 身份切换次数(IDSW)为0次, 帧率为27.65fps。该算法在复杂工业场景中兼顾了检测精度、跟踪鲁棒性与实时性, 为工业智能分拣提供了一种高效、可靠的目标跟踪解决方案。

     

    Abstract: To address the impact of common challenges such as small targets and dense arrangement in the 3C electronic component sorting scenario on the performance of multi-object tracking, proposes a lightweight multi-object tracking method based on improved YOLOv8n and ByteTrack. In the object detection stage, aiming at the difficulties in detecting small targets and mutual occlusion between targets, the original neck part of the YOLOv8n network is replaced with the Gold-YOLO module to enhance multi-scale feature fusion capability, and the ACmix attention mechanism is embedded in the N3 output layer of Gold-YOLO to strengthen the feature extraction ability for small and occluded targets. In the object tracking stage, algorithm parameters are optimized and a nested box filtering algorithm is added. It effectively suppresses mismatches caused by overlapping detection boxes and improves tracking robustness. The improved model's performance was verified using a self-built dataset. The experimental results show that the proposed improved YOLOv8n object detection algorithm and improved ByteTrack object tracking algorithm perform excellently in the electronic component sorting scenario. On the self-built dataset, the method achieved mAP@50 of 91. 32% in the object detection. In the object tracking, it can achieve the MOTA of 86. 72%, IDF1 of 93. 13%, number of identity switches (IDSW) of 0 times, and frame rate of 27. 65 fps. The proposed algorithm takes into account detection accuracy, tracking robustness and real-time performance in complex industrial scenarios, providing an efficient and reliable target tracking solution for industrial intelligent sorting scenarios.

     

/

返回文章
返回