feat: pct
本项目是基于Jittor框架实现的PCT模型,主要用于点云分类。 我使用了PCT模型,主要是因为Transformer类模型有较好的关联建模能力,能够更好地捕捉点云数据中的局部和全局特征,从而提高分类性能。 当前的网络结构仍然可以进行优化,同时可以在训练trick上有所改进,例如使用更好的数据增强方法、优化学习率调度等。
pip install -r requirements.txt
python pct.py
A Jittor implementation of Point Cloud Transformer(PCT) for ModelNet40 classification
版权所有:中国计算机学会技术支持:开源发展技术委员会 京ICP备13000930号-9 京公网安备 11010802032778号
PCT_jittor
本项目是基于Jittor框架实现的PCT模型,主要用于点云分类。 我使用了PCT模型,主要是因为Transformer类模型有较好的关联建模能力,能够更好地捕捉点云数据中的局部和全局特征,从而提高分类性能。 当前的网络结构仍然可以进行优化,同时可以在训练trick上有所改进,例如使用更好的数据增强方法、优化学习率调度等。
安装环境
使用方式