目录

PCT_jittor

A Jittor implementation of Point Cloud Transformer (PCT) for ModelNet40 classification.

Requirements

  • Python >= 3.7
  • Jittor >= 1.3.11
  • NumPy

Dataset

Download ModelNet40 dataset and place files in data/ folder:

  • train_points.npy
  • train_labels.npy
  • test_points.npy

Usage

Train the model:

```bash python pct.py –epochs 200 –batch_size 32 –output_name baseline

关于
30.0 KB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

版权所有:中国计算机学会技术支持:开源发展技术委员会
京ICP备13000930号-9 京公网安备 11010802047560号