目录

Point Cloud Transformer (PCT) - ModelNet40

基于 Jittor 框架实现的 Point Cloud Transformer (PCT) 基线代码,用于 ModelNet40 3D 点云分类任务。

项目结构

  • pct.py: 主程序脚本(包含数据集类、PCT 模型、训练与推理代码)
  • data/: 模型数据目录(包含训练和测试的点云及标签等数据,暂未纳入版本控制)
  • result.json: 模型测试集推理结果输出文件

快速开始

1. 环境依赖

pip install jittor numpy

2. 训练及验证

直接运行 python 脚本:

python pct.py

模型将自动在训练集上迭代,并在结束后输出 result.json

关于

This is a PA of the course Fundamentals of Computer Graphics in Tsinghua University

245.0 MB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

版权所有:中国计算机学会技术支持:开源发展技术委员会
京ICP备13000930号-9 京公网安备 11010802047560号