文件名称:Pointnet2.PyTorch:基于PyTorch的PointNet ++的更快实现
文件大小:74KB
文件格式:ZIP
更新时间:2024-06-03 11:48:03
Python
Pointnet2.PyTorch 基于的的PyTorch实现。 通过重新实现CUDA操作,比原始代码更快。 安装 要求 Linux(已在Ubuntu 14.04 / 16.04上测试) Python 3.6+ PyTorch 1.0 安装 通过运行以下命令来安装此库: cd pointnet2 python setup.py install cd ../ 例子 在这里,我提供了一个简单的示例,用于在KITTI Ourdoor前景点云分割任务中使用此库,有关任务描述和前景标签生成的详细信息,请参考论文 。 从网站下载训练数据,以下方式组织下载的文件: Pointnet2.PyTorch ├── pointnet2 ├── tools │ ├──data │ │ ├── KITTI │ │ │ ├── ImageSets │ │ │ ├── obj
【文件预览】:
Pointnet2.PyTorch-master
----tools()
--------pointnet2_msg.py(3KB)
--------data()
--------dataset.py(7KB)
--------train_and_eval.py(8KB)
--------_init_path.py(99B)
--------kitti_utils.py(8KB)
----LICENSE(1KB)
----pointnet2()
--------src()
--------pointnet2_modules.py(6KB)
--------setup.py(679B)
--------pointnet2_utils.py(9KB)
--------pytorch_utils.py(6KB)
----.gitignore(76B)
----README.md(2KB)