用视频扩散模型实现单目视频的快速3D密集跟踪
TrackCraft3R: Repurposing Video Diffusion Transformers for Dense 3D Tracking

- 将视频扩散模型改造成参考帧锚定的追踪框架
- 单次前向传播完成全视频跟踪,速度更快内存更少
- 适合需要高效长时序3D追踪的机器人与自动驾驶场景
从单目视频进行密集3D跟踪是动态场景理解的基础。尽管近期3D基础模型能提供可靠的逐帧几何信息,但在该几何中恢复物体运动仍具挑战性,需依赖真实视频学习的强运动先验。现有3D追踪器或采用从零训练的迭代范式,或微调基于静态多视角图像学习的3D重建模型,均缺乏真实世界运动先验。预训练的视频扩散变换器(video DiTs)来自互联网规模视频,具备丰富的时空先验,是3D追踪的有力基础。但其帧锚定生成方式与参考帧锚定的密集3D跟踪需求不匹配——后者需在时间上持续追踪同一物理点。我们提出TrackCraft3R,首个将video DiT直接复用于前馈式密集3D追踪的方法。给定单目视频及其帧锚定的重建点图,TrackCraft3R在一次前向传播中预测出参考帧锚定的追踪点图及可见性。通过两项设计实现:(i) 双隐变量表示,使用每帧几何隐变量和参考帧锚定的轨迹隐变量作为密集查询;(ii) 时间位置编码对齐,指定每个轨迹隐变量的目标时间戳。二者协同将video DiT的逐帧生成范式转化为参考帧锚定的追踪形式,支持LoRA微调。TrackCraft3R在标准稀疏与密集3D追踪基准上达到最先进性能,运行速度比最强先前方法快1.3倍,峰值内存减少4.6倍。还验证了对大位移和长视频的鲁棒性。
原文摘要 · Abstract (English)
Dense 3D tracking from monocular video is fundamental to dynamic scene understanding. While recent 3D foundation models provide reliable per-frame geometry, recovering object motion in this geometry remains challenging and benefits from strong motion priors learned from real-world videos. Existing 3D trackers either follow iterative paradigms trained from scratch on synthetic data or fine-tune 3D reconstruction models learned from static multi-view images, both lacking real-world motion priors. Pre-trained video diffusion transformers (video DiTs) offer rich spatio-temporal priors from internet-scale videos, making them a promising foundation for 3D tracking. However, their frame-anchored formulation, which generates each frame's content, is fundamentally mismatched with reference-anchored dense 3D tracking, which must follow the same physical points from a reference frame across time. We present TrackCraft3R, the first method to repurpose a video DiT as a feed-forward dense 3D tracker. Given a monocular video and its frame-anchored reconstruction pointmap, TrackCraft3R predicts a reference-anchored tracking pointmap that follows every pixel of the first frame across time in a single forward pass, along with its visibility. We achieve this through two designs: (i) a dual-latent representation that uses per-frame geometry latents and reference-anchored track latents as dense queries, and (ii) temporal RoPE alignment, which specifies the target timestamp of each track latent. Together, these designs convert the per-frame generative paradigm of video DiTs into a reference-anchored tracking formulation with LoRA fine-tuning. TrackCraft3R achieves state-of-the-art performance on standard sparse and dense 3D tracking benchmarks, while running 1.3x faster and using 4.6x less peak memory than the strongest prior method. We further demonstrate robustness to large motions and long videos.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。