diffusion model for single 3D mesh
- triplane encoder-decoder Done (based on NFD-tripane_decoder)
- diffusion model FIXME
- oct-tree optimize TODO
-
install pytorch and other requirements
pip install -r requirements.txt -
put xxx.obj file into 'dataset' folder
-
run command
python main.py --shape_name=xxxNote:
shape_nameshould be the same as input mesh file. If the input mesh is StoneWall.obj, then--shape_name=StoneWall -
output file can be found in 'output' folder
-
try diffuse in diffusion folder, run
python diffuse.pyafter put the triplane file into diffusion/data folder, and run visualize_diffused_shape.py to visualize.