PCT: Point Cloud Transformer for ModelNet40 Classification

基于 Jittor 框架实现的 PCT 点云分类模型，用于 ModelNet40 三维形状分类任务。

项目简介

本项目实现了 Point Cloud Transformer (PCT) 网络，完成 ModelNet40 点云分类任务：

• 输入：三维点云（1024 点）
• 输出：40 类形状分类结果
• 框架：Jittor
• 任务：训练 → 测试集推理 → 生成 result.json
• 输出文件格式:
  • pct_model.pkl：训练好的模型
  • result.json：测试集预测结果

运行指南

首先安装依赖：

pip install -r requirements.txt

接着将从https://cloud.tsinghua.edu.cn/f/f003de5a2e914d1e9e0e/?dl=1下载到的数据集解压到data/中

接着运行python文件即可

python pct.py

脚本支持参数：

• --data_dir：str 类型，数据集存放路径，默认值 ./data
• --n_points：int 类型，单样本点云采样点数，默认值 1024
• --batch_size：int 类型，训练批次大小，默认值 32
• --epochs：int 类型，训练总轮数，默认值 200
• --lr：float 类型，初始学习率，默认值 0.01
• --seed：int 类型，全局随机种子，默认值 42