This repo is a MindSpore implementation for Point-Transformer. Zhao et al.
python = 3.8
MindSpore = 1.7
Download dataset ModelNet40, save in
modelnet40_normal_resampled
python PointTransformerModelnet40Train.py
python PointTransformerModelnet40Eval.py
The accuracy of Point-Transformer in ModelNet40 is 0.918
Download ShapeNet, save in shapenetcore_partanno_segmentation_benchmark_v0_normal
.
python PointTransformerShapenetTrain.py
python PointTransformerShapenetEval.py
The inctance mIOU of Point-Transformer in Shapenet is 0.85
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