什么是 OpenPCDet?
OpenPCDet 是一个开源的 3D 目标检测框架,专注于点云数据处理,广泛应用于自动驾驶、机器人等领域。 78TP仓库:https://github.com/open-mmlab/OpenPCDet
Windows 安装前提条件
- Windows 10/11(推荐使用 WSL2 或原生支持)
- Python 3.7–3.10
- PyTorch ≥ 1.8(建议使用 CUDA 版本)
- CUDA 与 cuDNN(如使用 GPU)
- Visual Studio Build Tools(用于编译 C++ 扩展)
安装步骤
-
克隆仓库
git clone https://github.com/open-mmlab/OpenPCDet.git cd OpenPCDet -
创建虚拟环境(可选但推荐)
python -m venv openpcdet-env openpcdet-env\Scripts\activate -
安装 PyTorch(根据你的 CUDA 版本选择)
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -
安装依赖项
pip install -r requirements.txt -
编译扩展(spconv 等)
Windows 下需先安装
spconv-cu118(或其他对应版本):pip install spconv-cu118然后执行:
python setup.py develop
常见问题
- 编译失败? 确保已安装 Visual Studio 2019/2022 的 C++ 构建工具。
- 找不到 spconv? 推荐使用预编译 wheel,避免从源码构建。
- 路径中包含中文或空格? 请将项目放在纯英文路径下。
验证安装
运行以下命令测试是否安装成功:
python -c "import pcdet; print('OpenPCDet 导入成功!')"