目录

PCT_jittor

本项目是基于Jittor框架实现的PCT模型,主要用于点云分类。 我使用了PCT模型,主要是因为Transformer类模型有较好的关联建模能力,能够更好地捕捉点云数据中的局部和全局特征,从而提高分类性能。 当前的网络结构仍然可以进行优化,同时可以在训练trick上有所改进,例如使用更好的数据增强方法、优化学习率调度等。

安装环境

pip install -r requirements.txt

使用方式

python pct.py
关于

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

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

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