Pytorch Implementation of Various Point Transformers
Recently, various methods applied transformers to point clouds: PCT: Point Cloud Transformer (Meng-Hao Guo et al.), Point Transformer (Nico Engel et al.), Point Transformer (Hengshuang Zhao et al.). This repo is a pytorch implementation for these methods and aims to compare them under a fair setting. Currently, Point Transformer (Nico Engel et al.) is implemented.
Classification
Data Preparation
Download alignment ModelNet here and save in modelnet40_normal_resampled
.
Run
python train.py
Results
TBA
Miscellaneous
Some code and training settings are borrowed from https://github.com/yanx27/Pointnet_Pointnet2_pytorch.
TODOs
- implement Point Transformer (Hengshuang Zhao et al.)
- implement PCT: Point Cloud Transformer (Meng-Hao Guo et al.)