PCT_Pytorch

时间:2024-04-20 10:23:28
【文件属性】:

文件名称:PCT_Pytorch

文件大小:12.72MB

文件格式:ZIP

更新时间:2024-04-20 10:23:28

C++

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


【文件预览】:
PCT_Pytorch-main
----data_S3DIS.py(8KB)
----helper_ply.py(10KB)
----test.sh(142B)
----train.sh(132B)
----compile_op.sh(130B)
----utils()
--------nearest_neighbors()
--------cpp_wrappers()
--------data_prepare_semantic3d.py(4KB)
--------meta()
--------data_prepare_s3dis.py(4KB)
--------semantic-kitti.yaml(5KB)
--------download_semantic3d.sh(4KB)
--------data_prepare_semantickitti.py(4KB)
--------6_fold_cv.py(2KB)
----checkpoints()
--------best()
----helper_tool.py(13KB)
----README.md(1KB)
----pointnet2_ops_lib()
--------setup.py(1KB)
--------MANIFEST.in(29B)
--------pointnet2_ops()
----util.py(4KB)
----main_seg.py(9KB)
----main.py(9KB)
----model_seg.py(4KB)
----model.py(5KB)
----data.py(3KB)

网友评论