MMDetection环境配置(Window)

MMDetection环境配置(Window)

1. 安装和配置conda/miniconda

miniconda下载地址:https://docs.conda.io/en/latest/miniconda.html

在这里插入图片描述
根据提示安装好。
检查环境变量设置,需要根据自己conda/miniconda实际安装位置修改
在这里插入图片描述验证是否安装并配置成功:
在这里插入图片描述
可能还需要在cmd初始化conda:

conda init cmd.exe

2.配置虚拟环境

安装环境

创建名为“mmdet”的环境,使用python3.8(最好选择该版本,因为该版本有pycocotools-windows预编译包,更方便安装mmdet)

conda create -n mmdet python=3.8 -y

激活环境

conda activate mmdet

下载并安装pytorch

到官网上下载自己想要的pytorch,目前先推荐使用pytorch1.x版本。
如果慢的话可以尝试使用配置pip或conda镜像源方法加速,这里介绍的是离线安装的方法。

本人提供python3.8+pytorch1.8.1+cuda11.1的版本,可自行下载并安装。

链接:https://pan.baidu.com/s/1tsFF_7rED6szqFialymmUA
提取码:sdxx

pip install D:\whl\pyotrch1.8.1_win64_py38_cu111\torch-1.8.1+cu111-cp38-cp38-win_amd64.whl
pip install D:\whl\pyotrch1.8.1_win64_py38_cu111\torchvision-0.9.1+cu111-cp38-cp38-win_amd64.whl
pip install D:\whl\pyotrch1.8.1_win64_py38_cu111\torchaudio-0.8.1-cp38-none-win_amd64.whl

安装mmdetection

首先确认我们安装的是mmdetection2.x版本,暂不使用刚刚更新的3.x版本,不同版本mmdetection对mmcv-full有不同版本要求。
以下选择mmdetection2.x最后一个版本2.28.2,选择mmcv-full为1.6。

安装openmim,主要是为了方便安装mmcv-full

pip install openmim
mim install mmcv-full==1.6

推荐以下安装方法(可解决pycocotools无法编译安装的问题):
到mmdetection github上下载2.28.2版本:
https://github.com/open-mmlab/mmdetection/tree/v2.28.2
在这里插入图片描述
修改对应文件依赖,更改pycocotools为pycocotools-windows:
在这里插入图片描述
cd 进去mmdetection文件夹,进行安装:

python setup.py install

或者

python setup.py develop

注意这两个的差别,前者是将mmdet复制文件安装到虚拟环境上,作为一个不再修改的第三方库使用。后者是将当前安装的mmdet引用到虚拟环境中,可根据需求自己修改mmdet。

3.测试

运行mmdet的demo

python .\demo\image_demo.py .\demo\demo.jpg D:\code\mmdet_2.28.0_workstation\configs\solov2\solov2_light_r18_fpn_3x_coco.py D:\pretrained_model\mmdet\solov2_light_r18_fpn_3x_coco_20220511_083717-75fa355b.pth --out-file .\demo\demo_result.jpg

使用预训练的solov2模型推断demo图片,生成可视化。
在这里插入图片描述
完成!!接下来可以尝试跑训练了。

  • 2
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值