Jittor PCT Warmup

ModelNet40 点云分类的 Jittor 版本，基于官方 PCT 实现整理而来。

目录

• pct.py: 训练 / 推理入口
• networks/cls/pct.py: PCT 分类模型
• misc/ops.py: 官方点云算子

数据

把数据放到 data/ 下：

• train_points.npy
• train_labels.npy
• test_points.npy
• categories.txt

安装

pip install jittor numpy

训练

python pct.py --mode train --data_dir ./data --model_path checkpoints/pct_submit.pkl

生成结果

python pct.py --mode predict --data_dir ./data --model_path checkpoints/pct_submit.pkl --output result.json