PointNet.mxnet

This repo is implementation for PointNet(https://arxiv.org/abs/1612.00593) in mxnet. The model is in pointnet.py or models/. The implementation of this repo refers to this code

Download data and running

bash build.sh #build C++ code for visualization
bash download.sh #download dataset
python train_classification.py #train 3D model classification
python train_segmentation.py # train 3D model segmentaion

Performance

Without heavy tuning, PointNet can achieve 80-90% performance in classification and segmentaion on this dataset.

Sample segmentation result:

Links

• Pytorch implementation
• Project Page
• Tensorflow implementation