This work is based on existing studies: the generative model is based on this repo, the classification model is based on this repo. Please build the environment according to the corresponding requirements.
Usage:
- Train PointNet to have a classification model to be explained first, follow the step of this repo.
- Train our 3 different kinds of generative model via PointCloudAE.py, AED.py and NAED.py
- Run batch AM process on ModelNet40 for corresponding generative models via vanilla_VAE_AM_batch.py, AED_AM_batch.py and NAED_AM_batch.py