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

Multi-target floating garbage tracking algorithm for cleaning ships based on YOLOv5-Byte

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

     

    Abstract: Objectives Aiming at the problems of easy switching of tracking ID and low accuracy caused by complex scenes such as platform shaking and small target caused by long distance during the working process of cleaning ships, a multi-target tracking method based on improved YOLOv5-Byte was proposed. Methods Firstly, the Byte data association model was fused with the YOLOv5 detector to construct the multi-target tracking algorithm. Secondly, aiming at the problem that CIoU in YOLOv5 is sensitive to small targets, 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 targets. Finally, IoU is introduced into the adjustable hyperparameter α in the form of a power exponent in the Byte data association model to reduce the risk of small targets being discarded due to low confidence.Results The experimental results on the surface floating garbage data set showed that IDF1 and MOTA increased by 11.5% and 8.7% respectively, and IDs decreased for 7 times compared with the improvement.Conclusions The algorithm achieves accurate tracking of multiple small targets on the surface of the water, and provides a reference for the autonomous fishing technology of clean ships.

     

/

返回文章
返回