目录

PCT Jittor:基于 Jittor 实现的 Point Cloud Transformer 点云分类器

改进

  • 使用带权重衰减的 AdamW 替换基优化器 SGD。

  • 使用了更强的数据增强,包括随机旋转、随机缩放和随机平移。

使用方法

  • 将含有 train_points.npytrain_labels.npytest_points.npycategories.txt 四个文件的 data 文件夹(作为 ModelNet40的数据文件)放置在与 pct.py 同一个位置。

  • 运行 pct.py。通过指定 --batch_size--epochs--lr 指定训练参数。

关于

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

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

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