-
基于PyTorch的更快PointNet ++实现:Pointnet2.PyTorch
资源介绍
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
- 上一篇: openseg.pytorch:OCNet系列和SegFix的Pytorch代码
- 下一篇:没有了