Awesome

point-cloud-completion-shape-prior

This is the TensorFlow implementation for the paper "Point Cloud Completion by Learning Shape Priors"

Getting Started

python version: python-3.5; cuda version: cuda-10; TensorFlow version: tensorflow-1.13; Ubuntu version: 16.04

Compile Customized TF Operators

Please follow PointNet++ to compile customized TF operators. You may need to change the related library path if necessary.

Datasets

Ours
PCN's dataset

Train the model

To train the models: python train.py

Evaluate the models

Our pre-trained models can be downloaded here: Models, unzip and put them in the root directory.
To evaluate models: python test.py

Citation

@inproceedings{wang2020point,
     author = {Wang, Xiaogang and Ang Jr, Marcelo H and Lee, Gim Hee},
     title = {Point Cloud Completion by Learning Shape Priors},
     journal = {arXiv preprint arXiv:2008.00394},
     year = {2020}
}

Acknowledgements

Our implementations use the code from the following repository:
PCN
MMD-GAN