mmdetection环境配置,安装及demo检测

1.简介

mmdetection是一个基于PyTorch的开源对象检测工具箱,它提供了已公开发表的多种视觉检测核心模块,通过这些模块的组合,可以迅速搭建出各种著名的检测框架。

2.mmdetetion下载

https://github.com/open-mmlab/mmdetection

从上面给出的网址进行mmdetection压缩包的下载,下载完成后解压,在相应的文件夹里有一个docs文件夹,里面有一个get_started.md文件,打开get_started.md,里面是一些环境配置的要求和安装步骤。

 3.环境配置

首先我们创建一个虚拟环境,并激活该环境

conda create -n open-mmlab python=3.7 -y
source activate open-mmlab

安装torch环境

从get_started.md文件中我们可以看到其要求python版本要大于3.6,pytorch要大于1.3,cuda要大于9.2,这里我们进入PyTorch官网:https://pytorch.org/选择合适的版本进行命令安装。

例如我这里选用的是pytorch1.5,cuda10.1

# CUDA 10.1
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch

4.安装

首先安装mmcv,注意我们这里应当安装mmcv-full。同样是在get_started.md文件中我们可以看到安装mmcv-full的命令

由于pytorch和cuda版本不同,我输入的命令为

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html

安装mmdetection需要的扩展包

pip install -r requirements.txt

 这里需要注意:我们还要安装一个cocoAPI,在github官网直接搜索cocoAPI,选择第一个搜索结果进行压缩包下载,解压压缩包,进入里面的pythonAPI,执行以下命令进行安装

pip install -v -e.

如果你想验证自己是否安装成功,可以输入conda list进行查看。

conda list

如果出现pycocotools,则说明你的cocoAPI安装好了。

最后我们在mmdetection-master主文件夹里进行mmdetection的安装,执行以下命令。

python setup.py develop
#或者
pip install -v -e.

执行完成之后,同样我们可以输入conda list进行查看,如果出现mmdet则说明安装成功。

5.运行demo进行检测

 在运行demo之前,我们需要下载相应模型的权重文件,权重文件的下载链接我们可以在mmdetection-master/configs/faster-rcnn/README.md中可以找到(因为我这里选用的是faster-rcnn网络,选择其他的网络,进入configs里面的不同文件夹即可)

 选择你想使用的网络的权重文件,复制[model]后面的链接,打开链接可以直接进行下载,下载完成后在mmdetection-master里面新建一个文件夹checkpoints,将权重文件放入checkpoints文件夹里,完成之后输入以下命令:

python demo/image_demo.py demo/demo.jpg configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

 弹出上面的图片,说明你的mmdetection已经100%安装正确了。

  • 7
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 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、付费专栏及课程。

余额充值