Path Invariance Network in Tensorflow
- Two GPUs(GTX 1080 or better)
- Python 2.7
- Tensorflow 0.12.1
- Pointnet2 Customized TF Operators Please make sure the tf_ops folder is placed in the same directory.
First, download dataset here. (Point clouds are collected by Pointnet++) The pre-trained models can be downloaded here. (Point clouds models are trained using Pointnet++, and voxel models are trained using 3D-U-Net. The training code for voxel models will be released soon.)
To train models with downloaded dataset:
All training commands are included in the cmd file. Testing results are logged during the training stage.
Our code is released under BSD License (see LICENSE file for details).