Add model card for TrackCraft3R
#1
by nielsr HF Staff - opened
README.md
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
pipeline_tag: other
|
| 3 |
+
---
|
| 4 |
+
|
| 5 |
+
# TrackCraft3R: Repurposing Video Diffusion Transformers for Dense 3D Tracking
|
| 6 |
+
|
| 7 |
+
TrackCraft3R is the first method to repurpose a pre-trained video diffusion transformer (specifically Wan2.1-T2V-1.3B) as a single-pass dense 3D tracker. Given a monocular video along with its predicted depth and camera, TrackCraft3R predicts dense 3D trajectories in a single forward pass.
|
| 8 |
+
|
| 9 |
+
[**Paper**](https://huggingface.co/papers/2605.12587) | [**Project Page**](https://cvlab-kaist.github.io/TrackCraft3r) | [**GitHub**](https://github.com/cvlab-kaist/TrackCraft3r)
|
| 10 |
+
|
| 11 |
+
## Abstract
|
| 12 |
+
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. TrackCraft3R repurposes a pre-trained video diffusion transformer (video DiT) to capture rich spatio-temporal priors from internet-scale videos. By employing a dual-latent representation and temporal RoPE alignment, it converts the generative frame-anchored paradigm into a reference-anchored tracking formulation.
|
| 13 |
+
|
| 14 |
+
## Sample Usage
|
| 15 |
+
|
| 16 |
+
To run inference on a video, you first need to follow the environment setup and preprocessing steps (extracting depth and camera information) detailed in the [official repository](https://github.com/cvlab-kaist/TrackCraft3r). Once the input is prepared in `.npz` format, you can run inference as follows:
|
| 17 |
+
|
| 18 |
+
```bash
|
| 19 |
+
MODELSCOPE_CACHE=./checkpoints/wan_models \
|
| 20 |
+
python scripts/inference_user_video.py \
|
| 21 |
+
--checkpoint_path ./checkpoints/trackcraft3r/model.safetensors \
|
| 22 |
+
--input_npz ./breakdance_user.npz \
|
| 23 |
+
--output_npz ./breakdance_dense.npz \
|
| 24 |
+
--num_frames 12 --frame_stride 5
|
| 25 |
+
```
|
| 26 |
+
|
| 27 |
+
The output contains 3D tracks and visibility information, which can be visualized using the `visualize_dense.py` script provided in the repository.
|
| 28 |
+
|
| 29 |
+
## Citation
|
| 30 |
+
|
| 31 |
+
```bibtex
|
| 32 |
+
@article{nam2026trackcraft3r,
|
| 33 |
+
title={TrackCraft3R: Repurposing Video Diffusion Transformers for Dense 3D Tracking},
|
| 34 |
+
author={Nam, Jisu and Koo, Jahyeok and Son, Soowon and Jung, Jaewoo and An, Honggyu and Hur, Junhwa and Kim, Seungryong},
|
| 35 |
+
journal={arXiv preprint arXiv:2605.12587},
|
| 36 |
+
year={2026}
|
| 37 |
+
}
|
| 38 |
+
```
|