目录

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
邀请码