目录

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
邀请码