PCT-ModelNet40 (Jittor)

项目简介

• 项目名：PCT-ModelNet40 (Jittor)
• 简介：基于 Point Cloud Transformer (PCT) 的 ModelNet40 点云分类基线实现，使用 Jittor 框架。提供训练、预测与评测脚本，含内存友好的数据加载与推理选项。

主要文件

• pct.py：主训练/推理脚本，支持训练、加载已保存模型直接在测试集上生成预测 result.json 。
• data/：放置预处理好的数据文件，数据集下载网盘链接：https://cloud.tsinghua.edu.cn/d/cbac4255280f487296b6/ 。
• result.json：预测输出（被 .gitignore 忽略）。

依赖

建议使用虚拟环境，安装依赖：

pip install -r requirements.txt

快速开始

1. 训练模型（示例小规模快速跑）：

python pct.py --epochs 5 --batch_size 8 --n_points 512 --num_workers 0 --use_mmap

2. 使用已保存模型直接生成测试集预测（跳过训练）：

python pct.py --load_model pct_model.pkl --batch_size 8 --n_points 1024 --num_workers 0 --use_mmap

内存使用与运行建议

• 如果主机内存有限，推荐使用 --num_workers 0 与 --use_mmap 来降低内存峰值；同时减小 --batch_size 和 --n_points 可以显著降低内存占用。