PCT_jittor

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

环境安装

# 安装计图
conda create -n jittor python=3.9 -y
conda activate jittor
conda install -c conda-forge gcc=10 gxx=10 -y # 确保gcc、g++版本不高于10
conda install -c conda-forge libgomp -y # 确保OpenMP runtime存在

# 安装依赖
pip install jittor numpy

训练&推理

将数据放在 .data 目录下, 数据集下载见 计图挑战赛赛道二热身赛

# 运行指令
python pct.py