Linux (tested under Ubuntu 16.04 )
Python (tested under 2.7)
TensorFlow (tested under 1.4.0-GPU )
numpy, h5py
The code is built on the top of PointNET++ and PointConv. Before run the code, please compile the customized TensorFlow operators of PointNet++ under the folder "/Prediction_net/tf_ops".
To trian a model:
python -u run.py --mode=train --gpu=0
To test the trained model:
python -u run.py --mode=test --gpu=0