PCT_jittor

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

环境依赖

• Python 3.9+
• Jittor 1.3.11
• NumPy

数据集

ModelNet40 数据集可以从头歌平台下载。 将 train_points.npy、train_labels.npy、test_points.npy 放入 ./data/ 目录。

训练与测试

python pct.py

• 优化器：SGD (momentum=0.9, weight_decay=1e-4)
• 学习率：初始 0.01，采用带 5 epoch warmup 的余弦退火策略 (eta_min=1e-5)
• 数据增强：绕 Y 轴随机旋转

结果

ModelNet40 测试集分类准确率: 85.53%