/PointCNN.Pytorch

A PyTorch implemetation of PointCNN (CS294-131 Course Proj)

Primary LanguagePythonMIT LicenseMIT

PointCNN.PyTorch

This is a PyTorch implementation of PointCNN. It is as efficent as the origin Tensorflow implemetation and achieves same accuracy on both classification and segmentaion jobs. See the following references for more information:

"PointCNN"
Yangyan Li, Rui Bu, Mingchao Sun, Baoquan Chen
arXiv preprint arXiv:1801.07791, 2018.

https://arxiv.org/abs/1801.07791

Usage

We've tested code on ModelNet40 only.

python train_pytorch.py

License

Our code is released under MIT License (see LICENSE file for details).