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}
}