raniatze commited on
Commit
9c2674e
·
verified ·
1 Parent(s): 85d1a20

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. README.md +74 -0
  3. teaser.png +3 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ teaser.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ tags:
4
+ - 3d-scene-generation
5
+ - latent-diffusion
6
+ - autonomous-driving
7
+ - kitti-360
8
+ - primitives
9
+ - cvpr-2026
10
+ library_name: pytorch
11
+ ---
12
+
13
+ # PrITTI: Primitive-based Generation of Controllable and Editable 3D Semantic Urban Scenes
14
+
15
+ <p align="center">
16
+ <a href="https://drive.google.com/file/d/1Q7KgGbynzcIEyFJV1I17HgrYz6xrOwRJ/view?usp=sharing">📄 Paper</a> &nbsp;|&nbsp;
17
+ <a href="https://raniatze.github.io/pritti/">🌐 Project Page</a> &nbsp;|&nbsp;
18
+ <a href="https://github.com/autonomousvision/pritti">💻 Code</a>
19
+ </p>
20
+
21
+ <p align="center">
22
+ <img src="teaser.png" alt="PrITTI teaser" width="95%">
23
+ </p>
24
+
25
+ This repository hosts the **pre-trained checkpoints** for **PrITTI** (CVPR 2026), a latent-diffusion framework for controllable and editable 3D semantic urban scene generation, trained on [KITTI-360](https://www.cvlibs.net/datasets/kitti-360/).
26
+
27
+ ## Released Checkpoints
28
+
29
+ | File | Size | Description |
30
+ |------|------|-------------|
31
+ | `lvae.ckpt` | 1.1 GB | Layout Variational Autoencoder, trained for 300 epochs (`epoch=299, step=580200`). |
32
+ | `ldm_b/` | 773 MB | DiT-B Latent Diffusion Model in `diffusers`-pipeline format (`model_index.json` + `transformer/` + `decoder/` + `scheduler/`). |
33
+
34
+ ## Quick Start
35
+
36
+ Full environment setup, preprocessing, training, inference, and evaluation instructions live in the [official GitHub repository](https://github.com/autonomousvision/pritti). The snippet below downloads both checkpoints into the locations the code expects:
37
+
38
+ ```bash
39
+ # Make sure these are set (also documented in the main README)
40
+ export LVAE_TIMESTAMP="2025.06.03.17.23.30"
41
+ export LVAE_EPOCH="299"
42
+ export LVAE_STEP="580200"
43
+
44
+ # LVAE checkpoint
45
+ LVAE_DIR=$PRITTI_EXP_ROOT/exp/training_lvae_model/training_lvae_model/$LVAE_TIMESTAMP/checkpoints
46
+ mkdir -p $LVAE_DIR
47
+ huggingface-cli download raniatze/pritti-checkpoints lvae.ckpt --local-dir $LVAE_DIR
48
+ mv $LVAE_DIR/lvae.ckpt $LVAE_DIR/epoch=$LVAE_EPOCH-step=$LVAE_STEP.ckpt
49
+
50
+ # LDM (DiT-B) checkpoint
51
+ LDM_DIR=$PRITTI_EXP_ROOT/exp/training_dit_model/training_dit_b_model/training_dit_b_model/$LVAE_TIMESTAMP
52
+ mkdir -p $LDM_DIR
53
+ huggingface-cli download raniatze/pritti-checkpoints --include "ldm_b/*" --local-dir $LDM_DIR
54
+ mv $LDM_DIR/ldm_b $LDM_DIR/checkpoint
55
+ ```
56
+
57
+ Once downloaded, follow the [Inference](https://github.com/autonomousvision/pritti#-inference) section of the main README to reconstruct and generate scenes.
58
+
59
+ ## License
60
+
61
+ Released under **CC BY-NC 4.0** — free for academic and non-commercial research use. See [LICENSE](https://github.com/autonomousvision/pritti/blob/main/LICENSE) for full terms.
62
+
63
+ ## Citation
64
+
65
+ If you find PrITTI useful, please cite:
66
+
67
+ ```bibtex
68
+ @inproceedings{Tze2026PrITTI,
69
+ author = {Tze, Christina Ourania and Dauner, Daniel and Liao, Yiyi and Tsishkou, Dzmitry and Geiger, Andreas},
70
+ title = {PrITTI: Primitive-based Generation of Controllable and Editable 3D Semantic Scenes},
71
+ booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
72
+ year = {2026},
73
+ }
74
+ ```
teaser.png ADDED

Git LFS Details

  • SHA256: d0d60434ec03e2ebd7493b7086b38f0a490379fb119fe9ec20758e3933d5fe2d
  • Pointer size: 132 Bytes
  • Size of remote file: 1.66 MB