Add pipeline tag and enrich model card with installation and usage instructions

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +55 -3
README.md CHANGED
@@ -1,7 +1,9 @@
1
  ---
2
  license: mit
 
3
  ---
4
- # Orbis2: A Hierarchical World Model for Driving
 
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/), Jonannes Dienert, Rajat Sahay, [Thomas Brox](https://lmb.informatik.uni-freiburg.de/people/brox/index.html)
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
+ ![Teaser](https://github.com/lmb-freiburg/orbis2/raw/main/imgs/Framework.png)
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
+ ```