What it is

This repository is a pytorch version multi-gpu KP_Pyramid for point cloud data processing (some codes haven't been released now). We will release a clean version of code later, you can refer to a new version for training then.

How to train

For multi-gpu training

nohup python -u -m torch.distributed.launch --nproc_per_node=4 --master_port=$RANDOM train_dist.py --network pyramid_v2_deform --test-area-idx 0 >log_train_pyramid_v2_deform_s3dis_area0_0818.txt 2>&1 &

For single-gpu training

nohup python -u -m torch.distributed.launch --nproc_per_node=1 --master_port=$RANDOM train_dist.py --network pyramid_v2_deform --test-area-idx 0 >log_train_pyramid_v2_deform_s3dis_area0_0818.txt 2>&1 &

How to test

nohup python test_full_cloud.py result/kp_pyramid_v2_lr0p01_area0/ --dataset S3DIS --test-area-idx 0 --network pyramid_v2_deform --num-votes 50 >log_test_pyramid_v2_deform_s3dis_area0_0818_e500.txt 2>&1 &

Cite

@inproceedings{nie2022pyramid,
  title={Pyramid Architecture for Multi-Scale Processing in Point Cloud Segmentation},
  author={Nie, Dong and Lan, Rui and Wang, Ling and Ren, Xiaofeng},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={17284--17294},
  year={2022}
}

Our work uses part of codes from https://github.com/HuguesTHOMAS/KPConv-PyTorch. Thanks for the great work!