ModelNet40 Point Cloud Classification using PCT (Jittor)

本项目为计图比赛提供 ModelNet40 点云分类示例。

环境安装

conda create -n jittor python=3.9 -y
conda activate jittor
pip install --upgrade pip setuptools wheel
pip install "numpy<2" jittor
sudo apt update
sudo apt install -y gcc-12 g++-12
export cc_path=/usr/bin/g++-12
export CC=/usr/bin/gcc-12
export CXX=/usr/bin/g++-12

数据准备

数据应存放在 ./data 目录下，包含以下文件：

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

训练

python3.9 src/pct.py

评测/推理

运行训练后，生成的 result.json 可用于线上提交。

结果说明

• 评测指标为分类准确率（Accuracy），计算方式为正确分类样本数除以总样本数。