计算机工程与应用››2024,第60卷››问题(9): 122-134.内政部:10.3778/j.issn.1002-8331.2307-0359

•YOLOv8改进及应用专题 • 上一篇   下一篇

FA-SORT公司:轻量化的多车辆跟踪算法

欧阳博,朱勇建,杨礼康,王本源  

  1. 1浙江科技学院 机械与能源工程学院,杭州 310023
    2上海应用技术大学 计算机科学与信息工程学院,上海 401800
    三。宁波敏捷信息科技有限公司,浙江 宁波 315300
  • 出版日期:2024-05-01 发布日期:2024-04-29

FA-SORT:轻型多车辆跟踪算法

欧阳波、朱永建、杨利康、王本元  

  1. 1.浙江科技大学机械与能源工程学院,杭州310023
    2.上海理工大学计算机科学与信息工程学院,中国上海401800
    3.宁波民杰信息技术有限公司,中国浙江宁波315300
  • 在线:2024-05-01 出版:2024-04-29

摘要:近年来,无人机因体积小、灵活性好等优势被广泛应用在车辆跟踪领域。当无人机在高空飞行时,其捕捉的图像中车辆目标存在像素点少、拥挤以及被遮挡的情况。现有的多目标跟踪研究方法在车辆被遮挡过程中发生非线性运动时,使用卡尔曼滤波预测,会出现车辆位置预测不准确的问题。为了解决这些问题,采用先检测后跟踪(检测跟踪,待定)范式,对YOLOv8公司检测算法进行改进,在网络结构中引入了双成形器稀疏动态注意力模块,用于提取小目标特征信息。同时使用轻量级上采样算子卡拉夫替换原最近邻插值上采样,减少上采样过程中小目标特征丢失的问题。提出一种轻量化跟踪模型FA-SORT公司针对分拣算法提出三点改进:改进KF、YOLOv8公司算法。实验结果表明,与YOLOv8公司相比,精确率(精度)提高了0.97%,召回率(召回)提高了0.898%。对所提出的FA-SORT公司算法使用无人机DT数据集进行验证,结果表明,与现有的多目标跟踪算法相比,HOTA公司指标首个达到70.05%,IDF187.45%,跟踪速度达到29.93?FPS验证了FA-SORT公司跟踪算法在多车辆跟踪任务中的优越性。

关键词: 多车辆跟踪, 目标检测, 速度方向一致性匹配, 检测值匹配, 无人机图像

摘要:近年来,无人机因其体积小、灵活性强,在车辆跟踪领域得到了广泛的应用。然而,当无人机在高空飞行时,其捕获的图像中几乎没有像素点、拥挤和遮挡车辆对象。此外,现有的多目标跟踪研究方法在车辆遮挡过程中发生非线性运动时使用卡尔曼滤波预测,存在车辆位置预测不准确的问题。为了解决这些问题,本文采用检测跟踪(TBD)范式,首先通过在网络结构中引入BiFormer稀疏动态注意模块来改进YOLOv8检测算法,以提取小目标特征信息。同时,使用轻量级的上采样算子CARAFE来代替原来的最近邻插值上采样,减少了上采样过程中小对象特征丢失的问题。然后提出了一种轻量级跟踪模型FA-SORT,并对SORT算法提出了三点改进:改进KF、增加速度方向一致性匹配和检测值匹配。最后,在自制的多个车辆数据集组合上验证了改进的YOLOv8算法。实验结果表明,与YOLOv8相比,该算法的准确率提高了0.97%,召回率提高了0.898%。使用UAVDT数据集验证了所提出的FA-SORT算法,结果表明,与现有的多目标跟踪算法相比,第一HOTA度量达到70.05%,IDF1达到87.45%,跟踪速度达到29.93FPS。验证了FA-SORT跟踪算法在多车辆跟踪任务中的优越性。

关键词: 多车辆跟踪, 物体检测, 速度方向一致性匹配, 检测值匹配, 无人机图像