Official implementation of “DSPoint: Dual-scale Point Cloud Recognition with High-frequency Fusion”.
Paper link: https://arxiv.org/abs/2111.10332
Shape Classification of ModelNet-40 are given as an example of our method.
cd dspoint conda create -n dspoint python=3.7 conda activate dspoint conda install pytorch==1.6.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch pip install -r requirements.txt pip install pointnet2_ops_lib/.
Dataset will be automatically downloaded during training or testing.
Train your model with our default settings (same as the paper):
Evaluation will be done during training process.
Since the dataset is quite small (2468 for testing) and training performance on point cloud is quite random, it would be normal if you get model whose test accuracy varies between 93.0-93.5 (amount to 10 test data).
Evaluate with our pre-trained model (already included in ./checkpoints):
You should see the test accuracy at 93.48.
If you find this repo useful in your work or research, please cite our paper.
Our code borrows a lot from: