
PointNet2/PointNet++的PyTorch实现- Python项目开发
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本Python项目提供PointNet2(又称PointNet++)的PyTorch实现,适用于点云数据处理和机器学习任务,助力深度学习研究与应用开发。
用PyTorch编写的Pointnet2/Pointnet++的实现支持多GPU,并通过nn.DataParallel进行扩展。该版本适用于PyTorch 1.0及以上版本。对于较旧版本,可以参考v1.0分支中的代码。正式的模型定义和超参数可以在charlesq34/pointnet2仓库中找到(以tensorflow格式)。使用CUDA的GPU当前仅支持Pointnet++使用的自定义操作。
安装环境:已通过Python 3.6 和 Python 3.7 测试过此仓库。
全部评论 (0)
还没有任何评论哟~


