Home

Awesome

Neural Random Subspace (NRS), 3D Part

This repo includes the content that we use to verify the effectiveness of NRS (Neural Random Subspace) module on 3D Point Cloud Recognition Task.

NRS-diagram

It is based on the following implementations:

Major implementations of NRS can be found here: https://github.com/CupidJay/NRS_pytorch

For the details of ideas and results, please refer to our paper. We'd love you to cite it if you find it helpful :)

@article{NRS,
	title        = {Neural random subspace},
	author       = {Yun-Hao Cao and Jianxin Wu and Hanchen Wang and Joan Lasenby},
	year         = 2021,
	journal      = {Pattern Recognition},
	volume       = 112,
	pages        = 107801,
	doi          = {https://doi.org/10.1016/j.patcog.2020.107801},
	issn         = {0031-3203}
}

To start with:

- Data Preparation:
bash archive_bash/download_data.sh
- Training Models:
bash archive_bash/train_pointnet.sh
bash archive_bash/train_pointnet2.sh
bash archive_bash/train_dgcnn.sh
- FLOPS and #Params
see utils/FLOPs_Calculator.py for details
- Inference Time
bash archive_bash/timer.sh