8000 GitHub - NVlabs/PartPacker: Efficient Part-level 3D Object Generation via Dual Volume Packing
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

NVlabs/PartPacker

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files n 8000 avigation

PartPacker

teaser

This is the official implementation of PartPacker: Efficient Part-level 3D Object Generation via Dual Volume Packing.

Our model performs part-level 3D object generation from single-view images.

Install

We rely on torch with CUDA installed correctly.

pip install -r requirements.txt

Pretrained models

Download the pretrained models from huggingface, and put them in the pretrained folder.

mkdir pretrained
cd pretrained
wget https://huggingface.co/nvidia/PartPacker/resolve/main/vae.pt
wget https://huggingface.co/nvidia/PartPacker/resolve/main/flow.pt

Inference

# vae reconstruction of meshes
PYTHONPATH=. python vae/scripts/infer.py --ckpt_path pretrained/vae.pt --input assets/meshes/ --output_dir output/

# flow 3D generation from images
PYTHONPATH=. python flow/scripts/infer.py --ckpt_path pretrained/flow.pt --input assets/images/ --output_dir output/

# open local gradio app 
python app.py

Data Processing

We provide a Dual Volume Packing implementation to process raw glb meshes into two separate meshes as proposed in the paper.

cd data
python bipartite_contraction.py ./example_mesh.glb
# the two separate meshes will be saved in ./output

Acknowledgements

This work is built on many amazing research works and open-source projects, thanks a lot to all the authors for sharing!

Citation

@article{tang2024partpacker,
  title={Efficient Part-level 3D Object Generation via Dual Volume Packing},
  author={Tang, Jiaxiang and Lu, Ruijie and Li, Zhaoshuo and Hao, Zekun and Li, Xuan and Wei, Fangyin and Song, Shuran and Zeng, Gang and Liu, Ming-Yu and Lin, Tsung-Yi},
  journal={arXiv preprint arXiv:2506.09980},
  year={2025}
}

About

Efficient Part-level 3D Object Generation via Dual Volume Packing

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

0