This repo is implementation for PointNet(https://arxiv.org/abs/1612.00593) in pytorch. The model is in pointnet/model.py.
It is tested with pytorch-1.0.
git clone https://github.com/fxia22/pointnet.pytorch
cd pointnet.pytorch
pip install -e .
Download and build visualization tool
cd script
bash build.sh #build C++ code for visualization
bash download.sh #download dataset
Training
python main.py --dataset <dataset path> --nepoch=<number epochs>
Use --feature_transform to use feature transform.
On ModelNet40:
| Overall Acc | |
|---|---|
| Original implementation | 89.2 |
| this implementation(w/o feature transform) | 86.4 |
| this implementation(w/ feature transform) | 87.0 |