AED统一解决已知与未知目标跟踪,仅靠特征学习实现高鲁棒性。
Associate Everything Detected: Facilitating Tracking-by-Detection to the Unknown
- 将关联建模为相似性解码,融合空间、时间与跨片段相似度
- 在TAO、SportsMOT、DanceTrack上超越现有方法,无需先验知识
- 适合需要同时处理已知和未知类别跟踪的实用场景
多目标跟踪(MOT)是计算机视觉中的关键方向。传统闭集跟踪(CV-MOT)仅追踪预定义类别,而开放集跟踪(OV-MOT)虽能应对未知类别,但两类方法在对方任务中表现不佳。本文提出统一框架AED,兼容任意检测器,同时支持闭集与开放集跟踪,无需运动等先验知识。AED将关联任务建模为相似性解码问题,设计了融合空间、时间与跨片段相似度的sim-decoder,并引入关联中心学习机制,使特征具备连续跟踪能力且可泛化至未知类别。在TAO、SportsMOT和DanceTrack数据集上,AED在无先验条件下优于现有先进方法。代码已开源。
原文摘要 · Abstract (English)
Multi-object tracking (MOT) emerges as a pivotal and highly promising branch in the field of computer vision. Classical closed-vocabulary MOT (CV-MOT) methods aim to track objects of predefined categories. Recently, some open-vocabulary MOT (OV-MOT) methods have successfully addressed the problem of tracking unknown categories. However, we found that the CV-MOT and OV-MOT methods each struggle to excel in the tasks of the other. In this paper, we present a unified framework, Associate Everything Detected (AED), that simultaneously tackles CV-MOT and OV-MOT by integrating with any off-the-shelf detector and supports unknown categories. Different from existing tracking-by-detection MOT methods, AED gets rid of prior knowledge (e.g. motion cues) and relies solely on highly robust feature learning to handle complex trajectories in OV-MOT tasks while keeping excellent performance in CV-MOT tasks. Specifically, we model the association task as a similarity decoding problem and propose a sim-decoder with an association-centric learning mechanism. The sim-decoder calculates similarities in three aspects: spatial, temporal, and cross-clip. Subsequently, association-centric learning leverages these threefold similarities to ensure that the extracted features are appropriate for continuous tracking and robust enough to generalize to unknown categories. Compared with existing powerful OV-MOT and CV-MOT methods, AED achieves superior performance on TAO, SportsMOT, and DanceTrack without any prior knowledge. Our code is available at https://github.com/balabooooo/AED.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。