MMdetection3D学习系列(一)——环境配置安装

MMdetion3D是是mmlab在3d目标检测方面提供的相关检测模型,可以实现点云、图像或者多模态数据上的3D目标检测以及点云语义分割。
GitHub地址:https://github.com/open-mmlab/mmdetection3d/
目前mmdetection3d 支持21种不同的算法,100多个预训练模型,7个数据集:在这里插入图片描述
mmdetection3D安装比较简单,之前弄的时候踩了很多坑,老是卡在安装mmcv-full这一步,特别是在win系统下感觉老是有问题。在ubuntu系统下可以直接通过mim来安装很方便(windows 的 shell其实也可以,不过没搞透)。具体步骤如下(ubuntu系统或者win的shell):

创建conda环境

conda create -n mmdet3d python=3.7 pytorch==1.6.0 cudatoolkit=10.1 torchvision -c pytorch -y

这边要注意一点就是需要根据自身计算机的cuda版本来选择,这个很重要!!!
然后就是激活环境进行安装了,推荐使用mim进行安装,比较方便

conda activate mmdet3d
pip install openmim
mim install mmcv-full
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d-master
pip install -e .

需要注意的一点是有时候会在安装mmcv-full的时候出现问题。这个时候就需要到官网查看一下自己的cuda对应的版本了,我之前遇到过这情况,这么处理就可以了。
查看是否安装完成的话,可以在该环境下输入:

import mmdet3d
print(mmdet3d.__version__)

验证

可以在jupyter lab里进行验证。这里如果没有jupyter lab的话,直接pip install jupyterlab即可。
在命令行输入jupyter lab进入jupyterlab。在这里插入图片描述
输入如下指令即可:
在这里插入图片描述
其实就是调用mmdet3d的api进行检测,需要注意的是checkpoint需要自己去下载,可以在GitHub中的config目录下下载对应的model:
在这里插入图片描述
显示结果如下:在这里插入图片描述
如果是对点云的检测就是这样的:
在这里插入图片描述

  • 1
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
mmdetection是一个基于PyTorch框架的目标检测库,它提供了丰富的目标检测算法和实用工具。下面是mmdetection安装环境配置步骤: 1. 安装Anaconda:首先,确保你已经安装了Anaconda,可以通过官方网站下载对应的安装包进行安装。 2. 创建虚拟环境:打开Anaconda命令行工具或者终端,执行以下命令来创建一个新的虚拟环境: ``` conda create -n mmdet python=3.7 -y ``` 这将在当前目录下创建一个名为mmdet的虚拟环境,并使用Python 3.7版本。 3. 激活虚拟环境:创建完成后,执行以下命令来激活虚拟环境: ``` conda activate mmdet ``` 4. 安装依赖:在激活的虚拟环境中,执行以下命令来安装mmdetection所需的依赖: ``` conda install pytorch torchvision -c pytorch pip install mmcv-full==1.3.0 ``` 5. 下载mmdetection:你可以通过Git克隆mmdetection的代码库,执行以下命令来下载mmdetection: ``` git clone https://github.com/open-mmlab/mmdetection.git cd mmdetection ``` 6. 安装mmdetection:在mmdetection目录下,执行以下命令来安装mmdetection: ``` pip install -r requirements/build.txt pip install -v -e . ``` 7. 编译CUDA扩展:如果你的环境支持CUDA,并且你需要使用CUDA加速,可以执行以下命令来编译CUDA扩展: ``` pip install -r requirements/optional.txt cd mmdet/ops python setup.py develop cd ../.. ``` 至此,mmdetection安装环境配置完成。你可以通过运行示例代码或者参考mmdetection的文档来使用该库进行目标检测任务。
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值