window10+CPU安装配置mmdetection

1.mmdetection介绍

MMDetection是由open-mmlab开发的基于PyTorch的目标检测工具库open-mmlab是香港中文大学、商汤等机构联合成立的实验室。MMDetection 是一个目标检测工具箱,包含了丰富的目标检测、实例分割、全景分割算法以及相关的组件和模块,MMDetection 由 7 个主要部分组成,apis、structures、datasets、models、engine、evaluation 和 visualization。

MMdetection官方代码
官方说明文档

资源受限,贫苦学生党只有Window系统,只能用CPU,本文主要说明在Window下如何安装使用MMdetection框架

2.安装

2.1 安装配置conda环境

使用Conda创建并管理虚拟环境,Window10 下Conda的下载、安装和配置可参考window安装配置conda

conda create --name openmmlab python=3.8 -y
# 创建一个名为openmmlab的虚拟环境,python版本3.8
conda activate openmmlab
# 激活虚拟环境

2.2 安装PyTorch

安装 PyTorch时需要注意,官方文档在 CPU 平台上给出的命令:

conda install pytorch torchvision cpuonly -c pytorch

但是这行命令默认安装的是最新的PyTorch版本,后续我们安装MMCV时需要选择版本对应,而最新的MMCV的版本是2.0.0rc3 ,对应的Torch最高位1.9.x,所以在安装PyTorch时,应该指定版本。 对应关系可查看MMCV官方安装文档,如下图所示:(MMCV安装文档)

所以在安装PyTorch时,我们去PyTorch官方链接中安装指定版本。从上图可知,我们安装torch=1.9.x,安装mmcv=2.0.0rc。

在这里插入图片描述
点击红色部分,选择以前版本,找到Window CPU Only,以下是下载命令:

# CPU Only
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cpuonly -c pytorch

2.3 安装MMCV

MMDetection 可以在 CPU 环境中构建。在 CPU 模式下,可以进行模型训练(需要 MMCV 版本 >= 2.0.0rc1)、测试或者推理。不建议安装mmcv-lite版本,容易报错。有以上版本对应可知,下载命令如下:

pip install mmcv==2.0.0rc3 -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.9/index.html

2.4 安装 MMDetection

推荐使用Git进行克隆,同时也可以进行版本控制。Windw 下git安装教程

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
# "-v" 指详细说明,或更多的输出
# "-e" 表示在可编辑模式下安装项目,因此对代码所做的任何本地修改都会生效,从而无需重新安装。

3.验证安装

步骤 1. 下载配置文件和模型权重文件

mim download mmdet --config rtmdet_tiny_8xb32-300e_coco --dest .

完成后,你会在当前文件夹中发现两个文件 rtmdet_tiny_8xb32-300e_coco.py 和 rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth
直接运行以下命令进行验证:

python demo/image_demo.py \
               demo/demo.jpg  \
               rtmdet_tiny_8xb32-300e_coco.py \
               --weights rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth \
               --device cpu

在当前文件夹中的 outputs/vis 文件夹中看到一个新的图像 demo.jpg,图像中包含有网络预测的检测框。

  • 11
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值