PCT:点云变压器
这是PCT的Pytorch实施:点云变压器。
论文链接: :
要求
python> = 3.7
火炬> = 1.6
h5py
scikit学习
和
pip install pointnet2_ops_lib/.
该代码来自 和
楷模
我们在ModelNet40( )验证数据集上获得了93.2%的准确性
模型的路径在./checkpoints/train/models/model.t7中
培训和测试示例
# train
python main.py --exp_name=train --num_points=1024 --use_sgd=True --batch_size 32 --epochs 250 --lr 0.0001
# test
python main.py --exp_name=test --num_points=1024 --use_sgd
评论0
最新资源