Add pipeline tag and enrich model card with installation and usage instructions
#1
by nielsr HF Staff - opened
README.md
CHANGED
|
@@ -1,7 +1,9 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
| 3 |
---
|
| 4 |
-
|
|
|
|
| 5 |
|
| 6 |
<table>
|
| 7 |
<tr>
|
|
@@ -22,5 +24,55 @@ license: mit
|
|
| 22 |
</tr>
|
| 23 |
</table>
|
| 24 |
|
| 25 |
-
[Sudhanshu Mittal](https://lmb.informatik.uni-freiburg.de/people/mittal/), [Arian Mousakhan](https://lmb.informatik.uni-freiburg.de/people/mousakha/), [Silvio Galesso](https://lmb.informatik.uni-freiburg.de/people/galessos/), [Karim Farid](https://lmb.informatik.uni-freiburg.de/people/faridk/),
|
| 26 |
-
<br>University of Freiburg<br>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
pipeline_tag: image-to-video
|
| 4 |
---
|
| 5 |
+
|
| 6 |
+
# Orbis 2: A Hierarchical World Model for Driving
|
| 7 |
|
| 8 |
<table>
|
| 9 |
<tr>
|
|
|
|
| 24 |
</tr>
|
| 25 |
</table>
|
| 26 |
|
| 27 |
+
[Sudhanshu Mittal*](https://lmb.informatik.uni-freiburg.de/people/mittal/), [Arian Mousakhan*](https://lmb.informatik.uni-freiburg.de/people/mousakha/), [Silvio Galesso*](https://lmb.informatik.uni-freiburg.de/people/galessos/), [Karim Farid](https://lmb.informatik.uni-freiburg.de/people/faridk/), [Johannes Dienert](https://lmb.informatik.uni-freiburg.de/people/dienertj/), [Rajat Sahay](https://lmb.informatik.uni-freiburg.de/people/sahayr/), [Thomas Brox](https://lmb.informatik.uni-freiburg.de/people/brox/index.html)
|
| 28 |
+
<br>University of Freiburg<br>
|
| 29 |
+
<sub>* Main contributors</sub>
|
| 30 |
+
|
| 31 |
+
---
|
| 32 |
+
|
| 33 |
+
Orbis-2 is a hierarchical driving world model that generates long-horizon future video conditioned on past frames and an optional **steering signal**. A frozen low-frame-rate **L2** predictor provides abstract long-range context, while the **L1** detail predictor autoregressively generates high-frame-rate future frames. Steering can be given either as raw ego-motion values (speed and yaw rate) or as a 2D trajectory that the model should follow.
|
| 34 |
+
|
| 35 |
+

|
| 36 |
+
|
| 37 |
+
## Installation
|
| 38 |
+
```bash
|
| 39 |
+
git clone https://github.com/lmb-freiburg/orbis2.git
|
| 40 |
+
cd orbis2
|
| 41 |
+
conda env create -f environment.yml
|
| 42 |
+
conda activate orbis2_env
|
| 43 |
+
```
|
| 44 |
+
|
| 45 |
+
## Video Generation (Roll-out)
|
| 46 |
+
`evaluate/rollout_demo_v2.py` rolls out the world model from a single input video: it samples the L1 (high-rate) and L2 (low-rate, further back in time) context windows directly from the video, then autoregressively generates future frames.
|
| 47 |
+
|
| 48 |
+
Set the environment variable `ORBIS2_MODELS_DIR` with the path of the checkpoints folder, e.g.:
|
| 49 |
+
```bash
|
| 50 |
+
export ORBIS2_MODELS_DIR=./orbis2
|
| 51 |
+
```
|
| 52 |
+
|
| 53 |
+
To roll out the model using an input context video:
|
| 54 |
+
```bash
|
| 55 |
+
python evaluate/rollout_demo_v2.py \
|
| 56 |
+
--video /path/to/input_video.mp4 \
|
| 57 |
+
--output_dir rollout
|
| 58 |
+
```
|
| 59 |
+
|
| 60 |
+
To roll out with trajectory steering and an ego-centric trajectory overlay, specify an input trajectory file:
|
| 61 |
+
```bash
|
| 62 |
+
python evaluate/rollout_demo_v2.py \
|
| 63 |
+
--video /path/to/input_video.mp4 \
|
| 64 |
+
--trajectory_file trajectory.csv \
|
| 65 |
+
--vis_mode trajectory_ego \
|
| 66 |
+
--output_dir rollout_traj
|
| 67 |
+
```
|
| 68 |
+
|
| 69 |
+
## BibTeX
|
| 70 |
+
```bibtex
|
| 71 |
+
@article{orbis2_2026,
|
| 72 |
+
author = {Mittal, Sudhanshu and Mousakhan, Arian and Galesso, Silvio and
|
| 73 |
+
Farid, Karim and Dienert, Johannes and Sahay, Rajat and Brox, Thomas},
|
| 74 |
+
title = {Orbis 2: A Hierarchical World Model for Driving},
|
| 75 |
+
journal = {arXiv preprint arXiv:2607.15898},
|
| 76 |
+
year = {2026},
|
| 77 |
+
}
|
| 78 |
+
```
|