Folders and files

Name		Name	Last commit message	Last commit date
util		util
.gitignore		.gitignore
README.md		README.md
engine.py		engine.py
eval.py		eval.py
infer.py		infer.py
inverese.py		inverese.py
main.py		main.py
models.py		models.py
normalize.py		normalize.py
points.py		points.py

README

Geometry Distributions

Project Page | Paper (arXiv)

🚅 Training

torchrun --nproc_per_node=4 main.py --blr 5e-7 --output_dir output/loong --log_dir output/loong --data_path shapes/loong.obj

🎈 Inference

python infer.py --pth output/loong/checkpoint-999.pth --target Gaussian --num-steps 64 --output samples/loong --N 10000000

💾 Datasets

https://huggingface.co/datasets/Zbalpha/shapes

💼 Checkpoints

https://huggingface.co/Zbalpha/geom_dist_ckpt

📧 Contact

Contact Biao Zhang (@1zb) if you have any further questions. This repository is for academic research use only.

📘 Citation

arxiv

@article{zhang2024geometry,
  title={Geometry Distributions},
  author={Zhang, Biao and Ren, Jing and Wonka, Peter},
  journal={arXiv preprint arXiv:2411.16076},
  year={2024}
}

ICCV

@InProceedings{Zhang_2025_ICCV,
    author    = {Zhang, Biao and Ren, Jing and Wonka, Peter},
    title     = {Geometry Distributions},
    booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
    month     = {October},
    year      = {2025},
    pages     = {1495-1505}
}