win10下mmcv-full1.1.5与mmdet2.6的安装

win10下mmcv-full1.1.5与mmdet2.6的安装

一、先贴出本人基础环境配置:
Python 3.6、pytorch1.6.0、cuda10.1、opencv-python-4.2.0.34(opencv版本不一致可能使用中会报错)

二、先说为什么要安装mmcv-full1.1.5与mmdet2.6:
1.
mmcv-full与mmdet的开发者商汤貌似一开始并没有打算在windows上让用户使用这个库。查看https://download.openmmlab.com/mmcv/dist/index.html可以看到,大部分的mmcv-full版本只有linux下的whl包,只有mmcv-full1.1.3~1.1.5有几个win下的whl,这可能也是很多win用户在cmd中直接使用官方(https://github.com/open-mmlab/mmcv)推荐的mmcv-full安装命令:
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
会报错build不了whl的原因(个人推测)(本人在linux上使用此命令时,如果没有sudo,也会报这个错误)
介于以上原因,本人选择了mmcv-full1.1.5。
2.
在确定了mmcv-full1.1.5之后,便可以确定mmdet版本,在mmdet官方文档(https://mmdetection.readthedocs.io/en/latest/index.html)中可以查到二者的版本对应关系:
win10下mmcv-full1.1.5与mmdet2.6的安装_第1张图片

因此本人选择了mmdet2.6

三、开始正式安装:
①先安装mmcv-full的python依赖包:

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install -r requirements.txt

②安装mmcv-full:
https://download.openmmlab.com/mmcv/dist/index.html
win10下mmcv-full1.1.5与mmdet2.6的安装_第2张图片

可以看到官方支持的windows的mmcv-full安装包里,最高版本只到1.1.5

pip install mmcv-full==1.1.5+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html

或者离线安装:先下载好上图中支持win的whl安装包,再

pip install mmcv_full-1.1.5+torch1.6.0+cu101-cp36-cp36m-win_amd64.whl

③安装mmdet依赖环境

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements.txt

④安装mmdet2.6

pip install mmdet==2.6.0

至此,mmcv-full与mmdet包安装完成,但是在实际使用中一般还需要mmdet配套的的配置文件
win10下mmcv-full1.1.5与mmdet2.6的安装_第3张图片

同时在网上下载mmdet==2.6.0对应的完整mmdetection2.6文件(目前github上的为最新版,这个很坑,目前最新版里面的模型配置文件并不能适用于mmdet2.6):需要去commits中寻找历史版本,或去b站up小鸡炖技术的微信公众号与QQ群里找mmdetection2.6文件。 https://www.bilibili.com/video/BV1tA411j7UD?t=167

之后对mmcv.runner.epoch_based_runner.py的172行的shutil.copy(filepath,dst_file)修改,训练时注意使用mmdetection2.6里的配置文件。

四、用mmdetection2.6里自带的demo测试一下,结果如下:

大功告成!!!

你可能感兴趣的:(目标检测环境配置,pytorch,windows,系统安装)