Learning3D:3维点云数据的深度学习现代库

Learning3D是一个开源的Python库,专注于3D点云数据的深度学习算法。该库提供了多种预训练模型,如PointNet、PPFNet等,支持的数据集包括ModelNet40,并涵盖了点云分类、分割、配准等多种任务。此外,还包含各种损失函数,如Chamfer Distance和Correspondence Loss。库的使用包括数据加载、模型创建和损失函数应用等步骤,适用于不同的3D深度学习场景。
摘要由CSDN通过智能技术生成

Learning3D:3维点云数据的深度学习库

源链接:https://github.com/vinits5/learning3d

文档 | 博客 | 示例代码

Learning3D是一个开源库,支持处理3维数据的深度学习算法的开发。Learning3D有在python中目前效果好的深度神经网络。为了进一步开发,这里提供了模块化的代码。欢迎开源社区的贡献。

在Learning3D 可用的CV算法

序号 任务 算法
1 分类 PointNet, DGCNN, PPFNet, PointConv
2 分割 PointNet, DGCNN
3 重建 Point Completion Network (PCN)
4 配准 PointNetLK, PCRNet, DCP, PRNet, RPM-Net, DeepGMR
5 Flow Estimation FlowNet3D
6 Inlier Estimation MaskNet

可使用的预训练模型

  1. PointNet
  2. PCN
  3. PointNetLK
  4. PCRNet
  5. DCP
  6. PRNet
  7. FlowNet3D
  8. RPM-Net (clean-trained.pth, noisy-trained.pth, partial-pretrained.pth)
  9. DeepGMR
  10. PointConv (通过这个 链接下载)
  11. MaskNet

可使用的数据集

  1. ModelNet40

可使用的损失函数

  1. 分类损失函数(交叉熵(Cross Entropy))
  2. 配准损失函数 (FrobeniusNormLoss, RMSEFeaturesLoss)
  3. 距离损失函数(Chamfer Distance, Earth Mover’s Distance)
  4. 对应关系损失函数(Correspondence Loss) (基于这篇文章)

技术细节

支持的系统

  1. Ubuntu 16.04
  2. Ubuntu 18.04
  3. Linux Mint

开发环境

  1. CUDA 10.0 或更高的版本
  2. Pytorch 1.3 或更高的版本

如何使用这个库

请注意:要Clone 这个库到你的项目里,请不要将自己的代码添加到**learning3d**

  1. 所有的网络都定义在models这个模块中;
  2. 所有的损失函数都定义在losses这个模块中;
  3. Data loaders 提前定义在data_utils/dataloaders.py这个文件中;
  4. 所有的预训练模型都在learning3d/pretrained 文件夹之中;

文档

B: 批量大小(Ba

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值