arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.02147cs.CV
收藏

S2-Track:一种用于端到端3D多目标跟踪的简单而强大的方法

S2-Track: A Simple yet Strong Approach for End-to-End 3D Multi-Object Tracking

Shenzhen Campus of Sun Yat-sen University(中山大学深圳校区);Li Auto Inc.(理想汽车公司)

Tao Tang, Lijun Zhou, Pengkun Hao, Zihang He, Kalok Ho, Shuo Gu, Zhihui Hao, Haiyang Sun, Kun Zhan, Peng Jia, XianPeng Lang, Xiaodan Liang

AI总结:

针对端到端3D多目标跟踪在遮挡和小目标场景中的不足,本文从查询初始化、传播与匹配三方面改进,提出S2-Track,并在nuScenes上以66.3% AMOTA取得第一名。

摘要:

3D多目标跟踪(MOT)在自动驾驶感知中发挥着关键作用。近期基于查询的端到端跟踪器同时进行目标检测与跟踪,在3D MOT任务上展现出良好潜力。然而,现有方法仍处于发展早期,缺乏系统性改进,在某些复杂场景(如遮挡以及目标物体尺寸较小的情况)下无法有效跟踪目标。本文中,我们首先总结当前端到端3D MOT框架,将其分解为三个组成部分:查询初始化、查询传播和查询匹配。随后,我们提出相应改进,形成一个强大而简单的跟踪器:S2-Track。具体而言,在查询初始化方面,我们提出2D-Prompted Query Initialization(2D提示查询初始化),利用预测的2D目标和深度信息来提示对目标3D位置的初始估计。在查询传播方面,我们引入Uncertainty-aware Probabilistic Decoder(不确定性感知概率解码器),通过概率注意力捕捉复杂环境在目标预测中的不确定性。在查询匹配方面,我们提出Hierarchical Query Denoising(分层查询去噪)策略,以增强训练鲁棒性和收敛性。结果表明,我们的S2-Track在nuScenes基准上取得最先进性能,即在test划分上达到66.3% AMOTA,较此前最佳端到端方案显著高出8.9% AMOTA。我们在nuScenes跟踪任务排行榜上获得第一名。

英文摘要:

3D multiple object tracking (MOT) plays a crucial role in autonomous driving perception. Recent end-to-end query-based trackers simultaneously detect and track objects, which have shown promising potential for the 3D MOT task. However, existing methods are still in the early stages of development and lack systematic improvements, failing to track objects in certain complex scenarios, like occlusions and the small size of target object's situations. In this paper, we first summarize the current end-to-end 3D MOT framework by decomposing it into three constituent parts: query initialization, query propagation, and query matching. Then we propose corresponding improvements, which lead to a strong yet simple tracker: S2-Track. Specifically, for query initialization, we present 2D-Prompted Query Initialization, which leverages predicted 2D object and depth information to prompt an initial estimate of the object's 3D location. For query propagation, we introduce an Uncertainty-aware Probabilistic Decoder to capture the uncertainty of complex environment in object prediction with probabilistic attention. For query matching, we propose a Hierarchical Query Denoising strategy to enhance training robustness and convergence. As a result, our S2-Track achieves state-of-the-art performance on nuScenes benchmark, i.e., 66.3% AMOTA on test split, surpassing the previous best end-to-end solution by a significant margin of 8.9% AMOTA. We achieve 1st place on the nuScenes tracking task leaderboard.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑