开源PartPacker 3D，一张图片即可生成零件级三维3D模型

这是 PartPacker 的官方实现：通过双体积打包高效生成部件级 3D 对象。

我们的模型从单视图图像执行零件级 3D 对象生成。

GitHub:https://github.com/NVlabs/PartPacker

Demo:https://huggingface.co/spaces/nvidia/PartPacker

1.安装

我们依赖于正确安装 CUDA。torch

pip install -r requirements.txt

2.预训练模型

从 huggingface 下载预训练模型，并将其放入文件夹中。pretrained

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

3.推理

# 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

4.数据处理

我们提供了一个 Dual Volume Packing 实现，将原始 glb 网格处理成两个单独的网格，如论文中所建议的那样。

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

版权声明：本站所有文章版权均归AiNeuOS所有，任何个人、媒体、网站、团体等注明来源后均可以转载。但是不得在非我站的服务器上建立镜像，否则，我站将依法保留追究相关法律责任的权利。

关注公众号：