guochengqian/openpoints

OpenPoints: a library for easily reproducing point-based methods for point cloud understanding. The engine for [ASSANet](https://arxiv.org/abs/2110.10538), [PointNeXt](https://arxiv.org/abs/2206.04670)

解决的问题

OpenPoints 旨在提供一个标准化的库,用于基准测试和复现基于点的点云理解方法。它消除了从头编写复杂点云操作的需要,使研究人员能够公平地比较不同模型,并更轻松地复现结果。

工作原理

该库作为一个引擎,提供了一系列基础操作,如图卷积、自注意力、最远点采样和球查询,这些操作被用于构建各种点云网络。它采用基于配置的系统(YAML 文件),无需更改代码即可轻松构建模型、优化器、调度器和损失函数。

适用人群

主要面向从事 3D 点云理解与计算机视觉研究的科研人员和开发者,特别是使用 PyTorch 生态系统的用户。

主要亮点

  • 广泛模型支持:支持复现多种代表性网络,包括 PointNet、DGCNN、DeepGCN、PointNet++、ASSANet、PointMLP、PointNeXt、Pix4Point 和 PointVector。
  • 配置驱动:只需修改 YAML 配置文件,即可在不同任务上训练和验证不同模型。
  • 可扩展框架:提供一套完整的基础操作,便于构建新网络。
  • CUDA 加速:包含专为高性能点云处理设计的 C++/CUDA 操作符。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目