Update ModelNet40 data & prediction results
使用带权重衰减的 AdamW 替换基优化器 SGD。
使用了更强的数据增强,包括随机旋转、随机缩放和随机平移。
将含有 train_points.npy,train_labels.npy,test_points.npy 和 categories.txt 四个文件的 data 文件夹(作为 ModelNet40的数据文件)放置在与 pct.py 同一个位置。
train_points.npy
train_labels.npy
test_points.npy
categories.txt
data
pct.py
运行 pct.py。通过指定 --batch_size,--epochs 和 --lr 指定训练参数。
--batch_size
--epochs
--lr
A Jittor implementation of Point Cloud Transformer (PCT) for ModelNet40 classification
版权所有:中国计算机学会技术支持:开源发展技术委员会 京ICP备13000930号-9 京公网安备 11010802047560号
PCT Jittor:基于 Jittor 实现的 Point Cloud Transformer 点云分类器
改进
使用带权重衰减的 AdamW 替换基优化器 SGD。
使用了更强的数据增强,包括随机旋转、随机缩放和随机平移。
使用方法
将含有
train_points.npy,train_labels.npy,test_points.npy和categories.txt四个文件的data文件夹(作为 ModelNet40的数据文件)放置在与pct.py同一个位置。运行
pct.py。通过指定--batch_size,--epochs和--lr指定训练参数。