feat: implement data enhancing features
基于 Jittor 框架实现的 Point Cloud Transformer (PCT) 基线代码,用于 ModelNet40 3D 点云分类任务。
pct.py
data/
result.json
pip install jittor numpy
直接运行 python 脚本:
python pct.py
模型将自动在训练集上迭代,并在结束后输出 result.json。
This is a PA of the course Fundamentals of Computer Graphics in Tsinghua University
版权所有:中国计算机学会技术支持:开源发展技术委员会 京ICP备13000930号-9 京公网安备 11010802047560号
Point Cloud Transformer (PCT) - ModelNet40
基于 Jittor 框架实现的 Point Cloud Transformer (PCT) 基线代码,用于 ModelNet40 3D 点云分类任务。
项目结构
pct.py: 主程序脚本(包含数据集类、PCT 模型、训练与推理代码)data/: 模型数据目录(包含训练和测试的点云及标签等数据,暂未纳入版本控制)result.json: 模型测试集推理结果输出文件快速开始
1. 环境依赖
2. 训练及验证
直接运行 python 脚本:
模型将自动在训练集上迭代,并在结束后输出
result.json。