3 个月前

StrongSORT:让DeepSORT重放光彩

StrongSORT:让DeepSORT重放光彩

摘要

近年来,多目标跟踪(Multi-Object Tracking, MOT)受到广泛关注,相关研究取得了显著进展。然而,现有方法普遍采用多种基础模型(如检测器和特征嵌入模型),并引入不同的训练或推理技巧,导致难以构建一个公平、可靠的基准进行对比。为此,本文首先对经典跟踪器DeepSORT进行了重新审视,并从目标检测、特征嵌入和轨迹关联等多个方面进行了显著改进,提出了一种新型跟踪器——StrongSORT。该方法为MOT领域提供了强大且公平的基准。此外,本文还提出了两种轻量级、即插即用的算法,以解决MOT中两个固有的“缺失”问题:关联缺失与检测缺失。具体而言,不同于大多数方法在高计算复杂度下将短轨迹片段关联为完整轨迹,本文提出了一种无需外观信息的全局关联模型——AFLink(Appearance-Free Link),可在不依赖外观特征的情况下实现高效、准确的全局轨迹关联,在速度与精度之间取得了良好平衡。同时,本文提出一种基于高斯过程回归的高斯平滑插值方法(Gaussian-smoothed Interpolation, GSI),用于缓解因目标丢失导致的检测缺失问题。AFLink与GSI可轻松集成至各类跟踪器中,额外计算开销极低(在MOT17数据集上,每帧分别仅增加1.7毫秒和7.1毫秒)。最终,通过将StrongSORT与AFLink和GSI融合,构建出的强跟踪器StrongSORT++在多个公开基准测试集上(包括MOT17、MOT20、DanceTrack和KITTI)均取得了当前最优性能。相关代码已开源,分别发布于:https://github.com/dyhBUPT/StrongSORT 和 https://github.com/open-mmlab/mmtracking。

基准测试

基准方法指标
multi-object-tracking-on-mot17StrongSORT
HOTA: 64.4
IDF1: 79.5
MOTA: 79.6
multi-object-tracking-on-mot20-1StrongSORT
HOTA: 62.6
IDF1: 77.0
MOTA: 73.8

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
StrongSORT:让DeepSORT重放光彩 | 论文 | HyperAI超神经