pip install pointnet2_ops_lib/.
NOTE: If you encounter problems while building the kernel, you can refer to Pointnet2_PyTorch for solutions.
The IntrA dataset can be downloaded from intra3d2019, and you need to unzip the files to data/IntrA3D
.
The ModelNet40 dataset is automatically downloaded.
For IntrA model train (1024 points)
CUDA_VISIBLE_DEVICES=xx python main_intra.py --exp_name intra_cls_train --mode train --num_points 1024 --lr 0.01 --use_sgd True --num_K 32 64
For ModelNet40 model train (1024 points)
CUDA_VISIBLE_DEVICES=xx python main_m40.py --exp_name m40_cls_train --mode train --num_points 1024 --epochs 250 --lr 0.001 --num_K 20 20
NOTE: To achieve a fast computational speed, you can also uncomment torch.backends.cudnn.benchmark = True
and comment out torch.backends.cudnn.deterministic = True
, while the final results might vary.
Our code borrows from: