Windows下安装mmdet
MMDetection:是一个基于 PyTorch 的目标检测开源工具箱。但是依赖 mmcv,支持 pytorch 1.5+。
mmcv:类似OpenCV,是一个开源的面向计算机视觉的基础库,高质量实现CUDA算子,使用它的话需要 python 3.6+
所需软件(包)介绍
安装vs studio2019 “必须是2019,2022我试过不行”并且要在安装的时候勾选C++桌面编译条件 用里面的一个编译软件去编译文件,用于安装mmcv,因为mmcv没有提供编译好的可用于windows的文件
VS 2019 建议安装2019版本。
MMDetection:2.20.0 (地址:https://github.com/open-mmlab/mmdetection/blob/master/docs/en/get_started.md)
mmcv:1.3.17 (参考:https://github.com/open-mmlab/mmdetection/blob/master/docs/en/get_started.md)由 VS的编译工具 cl 编译得到。
注:想选择使用2.20.0版本,但是最后你会发现没办法使用,他会退回到2.11.0,Github上issues中解决办法就会让他退回到2.11.0这个版本。
添加环境变量到系统环境变量->Path下,地址在安装目录下(根据自己的安装位置):D:\install_in_D\VS2019\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx86\x64
完成后打开 cmd 测试安装是否成功
输入以下两条命令任意一条都可:
conda -V
conda --version
更新 conda 包(此步骤非必须,花费时间比较久)
conda upgrade –all
(1)查看conda所有环境
conda info -e
或
conda env list
(2)创建新环境
conda create -n your_env_name
方案二(指定目录创建)
指定目录创建:如在 E:\workspace\Python\Pytorch\STOD 目录下执行下面的命令:
执行:conda create --prefix=E:\workspace\Python\Pytorch\STOD\swin_det python=3.8
激活 conda 环境:activate E:\workspace\Python\Pytorch\STOD\swin_det
创建指定python版本的环境
conda create --name your_env_name python=2.7
conda create -n your_env_name python=3
conda create -n your_env_name python=3.7
(3)进入环境
conda activate your_env_name
(4)退出当前环境
conda deactivate
(5)删除环境
conda remove -n your_env_name --all
# 删除所有安装包及 cache
conda clean -y --all
(6)克隆环境
conda create -n new_env_name -c old_env_name
(7)查看某个环境下安装的包
conda list
安装 mmdetedtion 环境
1、创建 open-mmlab 虚拟环境
打开 Anaconda Prompt(anaconda版的cmd 命令提示窗,跟随 anaconda 自动安装在电脑 上),输入命令:
conda create -n open-mmlab python=3.7 -y
2、进入刚刚创建好的环境
cmd 下:
activate open-mmlab
Anaconda Prompt 下:
conda activate open-mmlab
3、安装 CUDA10.1 和 pytorch
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch -y
conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch -y
pip3 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
指定阿里源 (国内一般需要)-i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
安装其他必要的Python包
在虚拟环境下执行:
pip install cython matplotlib opencv-python timm
4、安装 mmcv
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
打开 conda的power shell(一定要用这个)
进入到 conda环境目录,激活conda 环境
直接创建的conda环境的使用:conda activate swin_det
指定了conda环境路径的使用:conda activate .\mmdet_swin
验证:codna list 会看到 mmcv-full(记得确定一下版本对不对 )
5、安装 MMDetection
:MMDetection 官网
git clone https://github.com/open-mmlab/mmdetection.git
# 或 直接去官网下载压缩包 不用解压
cd mmdetection # 进入mmdetection目录
pip install -r requirements/build.txt
pip install -v -e .