This repository TreeGAN is for 3D Point Cloud Generative Adversarial Network Based on Tree Structured Graph Convolutions paper accepted on ICCV 2019
3D Point Cloud Generative Adversarial Network Based on Tree Structured Graph Convolutions
(Dong Wook Shu*, Sung Woo Park*, Junseok Kwon)
TreeGAN network consists of "TreeGCN Generator" and "Discriminator".
For more details, refer our paper.
-
Multi Class Generation.
-
Single Class Generation.
-
Single Class Interpolation.
-
This FPD version is used pretrained PointNet.
-
This FPD version is for ShapeNet-Benchmark dataset from A Scalable Active Framework for Region Annotation in 3D Shape Collections.
-
Our pretrained PointNet-FPD version use only subset of official ShapeNet dataset to get PointNet classification performance higher than 95%.
inproceedings{~~, title={}, author={}, year={2019} }
This project was tested on Windows 10 / Ubuntu 16.04 Using conda install command is recommended to setting.
- Python 3.6
- Numpy
- Pytorch 1.0
- visdom
In our project, arguments.py file has almost every parameters to specify for training.
For example, if you want to train, it needs to specify dataset_path argument.