简介
OpenPCDet是一个用于3D目标检测的开源工具箱,它提供了多种数据集的加载器,支持多种模型,并且易于扩展。在本博客中,我将引导你完成OpenPCDet的安装过程,并展示如何测试和训练预训练模型。
安装OpenPCDet
环境要求
在开始之前,请确保你的系统满足以下要求:
- Linux操作系统(Ubuntu 14.04/16.04/18.04/20.04/21.04已测试)
- Python 3.6或更高版本
- PyTorch 1.1或更高版本(测试了1.1, 1.3, 1.5到1.10)
- CUDA 9.0或更高版本(PyTorch 1.3+需要CUDA 9.2+)
- spconv库(v1.0, v1.2或v2.x)
安装步骤
-
克隆仓库:
git clone https://github.com/open-mmlab/OpenPCDet.git #官网测试的是pcdet v0.5 #可以直接使用master进行编译,等其出错再说。 #使用如下命令进行切换 git tag #v0.1.0 #v0.1.1 #v0.2.0 #v0.3.0 #v0.5.0 #v0.5.2 git checkout v0.5.2
-
安装依赖库:
- 安装SparseConv库。根据你的PyTorch版本选择合适的spconv版本。
- 对于PyTorch 1.1,安装spconv v1.0(commit 8da6f96)。
- 对于PyTorch 1.3+,安装spconv v1.2或使用作者推荐的Docker。
- 或者,你可以使用pip安装最新的spconv v