Skip to content

anc2001/SP-GAN

 
 

Repository files navigation

Tree Generation with SP-GAN

Project procedurally generates trees and trains SP-GAN to generate them.

Refer to original README for model usage

Procedural Tree Generation

Trees were procedurally generated with this blender addon

See the scripts in dataset_generation for dataset creation. generate_tree_dataset.py is a blender script to export the tree meshes and convert_dataset.py will convert them to pointclouds.

Two datasets were generated. One only uses the acer tree preset in the blender addon and the other uses all 20 preset types. They are available here

Training and Results

Two models were trained for 300 epochs. Some results are shown below. For training graphs and more examples see results. The pretrained models can be found here

Acer

Part Correspondence

Shape Interpolation

All 20 Trees

Part Correspondence

Shape Interpolation

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 77.7%
  • Cuda 15.1%
  • C++ 6.5%
  • Other 0.7%