基于YOLOv5-Byte面向清洁船的多目标漂浮垃圾追踪算法

A multi-object tracking algorithm for cleaning vessels to track floating debris based on YOLOv5-Byte

  • 摘要:
    目的 针对清洁船在自主打捞作业过程中因平台晃动、距离较远造成的目标小等复杂场景引起的追踪准确率低、目标身份编号(ID) 易切换等问题,提出一种基于改进YOLOv5-Byte的多目标漂浮垃圾追踪算法。
    方法 首先,引入Byte数据关联模型与YOLOv5检测器融合,实现多目标追踪(MOT)算法的构建;其次,针对YOLOv5中完整交并比(CIoU)对小目标敏感的问题,对边界框高斯建模,提出归一化Wasserstein距离度量;然后,引入平衡因子来调节CIoU和归一化Wasserstein距离度量在损失函数中的贡献度,并调节检测器对小目标敏感度;最后,在Byte数据关联模型中将交并比(IoU)以幂指数形式引入可调节超参数,降低目标因关联值低而被丢弃的风险。
    结果 基于水面漂浮垃圾数据集的实验结果表明,与算法改进前相比,识别平均数比率(IDF1)提高11.5%、多目标追踪准确率(MOTA)提高8.7%,目标身份编号切换次数(IDs)下降7次。
    结论 所提算法实现了水面多个小目标的准确追踪,为清洁船实现自主打捞技术提供了参考。

     

    Abstract:
    Objective Aiming at the problems of easy switching of tracking ID (identity) and low accuracy caused by complex scenes, including platform shaking and small object due to long distance during autonomous collection of surface floating debris by cleaning vessels, a multi-object tracking (MOT) method based on improved YOLOv5-Byte was proposed.
    Method First, the Byte data association model was integrated with the YOLOv5 detector to construct the MOT algorithm. Secondly, aiming at the problem that CIoU (complete intersection over union) in YOLOv5 is sensitive to small objects, the normalized Wasserstein distance metric is proposed for the boundary box Gaussian modeling. Then, the balance factor is introduced to adjust the contribution of CIoU and normalized Wasserstein distance measures to the loss function to adjust the sensitivity of the detector to small objects. Finally, IoU (intersection over union)is introduced into the adjustable hyper-parameter in the form of a power exponent in the Byte data association model to reduce the risk of small objects being discarded due to low confidence.
    Results  The experimental results on the surface floating debris dataset showed that the tracking assessment metrics IDF1 (identification F1 score) and MOTA (multiple object tracking accuracy) increased by 11.5% and 8.7%, respectively, while the number of IDs (identity switches) decreased by 7 compared to the algorithm before improvement.
    Conclusion The proposed algorithm achieves accurate tracking of multiple small objects on the water surface, providing a reference for the autonomous debris collection technology of intelligent floating cleaning vessels.

     

/

返回文章
返回