Point-Transformer-Mindspore

This repo is a MindSpore implementation for Point-Transformer. Zhao et al.

Requiremernt

python = 3.8
MindSpore = 1.7

Classification

Download dataset ModelNet40, save in modelnet40_normal_resampled

Train

python PointTransformerModelnet40Train.py

Eval

python PointTransformerModelnet40Eval.py

Result

The accuracy of Point-Transformer in ModelNet40 is 0.918

Part Segmentation

Download ShapeNet, save in shapenetcore_partanno_segmentation_benchmark_v0_normal.

Train

python PointTransformerShapenetTrain.py

Eval

python PointTransformerShapenetEval.py

Result

The inctance mIOU of Point-Transformer in Shapenet is 0.85

Acknowledgements

Our implementation is mainly based on the following codebases. We gratefully thank the authors for their wonderful works. qq456cvb, Point-Transformer; HelloMaroon, Point-Transformer-classification; pierrefdz,point-transformer