面向点云的开源 3D 目标检测框架
OpenPCDet 是由 OpenMMLab 社区维护的一个开源、模块化、高性能的 3D 目标检测框架, 专注于处理来自 LiDAR 等传感器的点云数据。它支持多种主流 3D 检测模型,并提供统一的数据接口和训练流程, 极大简化了研究与开发工作。
解耦数据预处理、模型结构、损失函数等组件,便于扩展与复用。
内置 PointRCNN、PartA2、PV-RCNN、CenterPoint 等经典算法。
基于 PyTorch,支持分布式训练与混合精度加速。
提供清晰文档、预训练模型与一键训练脚本。
安装 OpenPCDet 只需几行命令:
git clone https://github.com/open-mmlab/OpenPCDet.git
cd OpenPCDet
pip install -r requirements.txt
python setup.py develop
训练一个 PV-RCNN 模型:
python train.py --cfg_file cfgs/kitti_models/pv_rcnn.yaml