文件名称:PCT_Pytorch:PCT的Pytorch实施
文件大小:10.23MB
文件格式:ZIP
更新时间:2024-06-02 13:09:19
Python
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_sg
【文件预览】:
PCT_Pytorch-main
----util.py(4KB)
----checkpoints()
--------best()
----data.py(3KB)
----model_new.py(5KB)
----main.py(9KB)
----model.py(5KB)
----train.sh(132B)
----pointnet2_ops_lib()
--------MANIFEST.in(29B)
--------pointnet2_ops()
--------setup.py(1KB)
----LICENSE(1KB)
----test.sh(142B)
----README.md(1KB)