Home

Awesome

3DMedPT

<!-- **\[our code for IntrA and Modelnet40 classification is released]** 3D Medical Point Transformer: Introducing Convolution to Attention Networks for Medical Point Cloud Analysis [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/3d-medical-point-transformer-introducing/classification-on-intra)](https://paperswithcode.com/sota/classification-on-intra?p=3d-medical-point-transformer-introducing) [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/3d-medical-point-transformer-introducing/3d-part-segmentation-on-intra)](https://paperswithcode.com/sota/3d-part-segmentation-on-intra?p=3d-medical-point-transformer-introducing) [[arxiv]](https://arxiv.org/pdf/2112.04863.pdf) [[project page]](https://3dmedpt.github.io/) Author: Jianhui Yu, Chaoyi Zhang, Heng Wang, Dingxin Zhang, Yang Song, Tiange Xiang, Dongnan Liu, Weidong Cai --> <!-- ## Model Architecture ![model architecture](./images/model_details.jpg) -->

Requirements

Data

The IntrA dataset can be downloaded from intra3d2019, and you need to unzip the files to data/IntrA3D.

The ModelNet40 dataset is automatically downloaded.

Performance

Training Command

<b>NOTE:</b> To achieve a fast computational speed, you can also uncomment torch.backends.cudnn.benchmark = True and comment out torch.backends.cudnn.deterministic = True, while the final results might vary.

<!-- ## Citation If you find our data or project useful in your research, please cite: ``` @article{yu20213d, title={3D Medical Point Transformer: Introducing Convolution to Attention Networks for Medical Point Cloud Analysis}, author={Yu, Jianhui and Zhang, Chaoyi and Wang, Heng and Zhang, Dingxin and Song, Yang and Xiang, Tiange and Liu, Dongnan and Cai, Weidong}, journal={arXiv preprint arXiv:2112.04863}, year={2021} } ``` -->

Acknowledgement

Our code borrows from: