windows安装MMDetection并测试

爱运动的IT小白

已于 2023-06-02 23:09:33 修改

阅读量633

点赞数

文章标签： python pytorch 深度学习

于 2023-06-02 20:44:35 首次发布

本文链接：https://blog.csdn.net/weixin_53297048/article/details/131013474

版权

Anaconda中创建虚拟环境

1、创建python=3.9.16,pytorch-cuda=11.8,pytorch=2.0.1的虚拟环境并下载GPU版pytorch

//创建虚拟环境
conda create --nam openmmlab python=3.9 -y
conda activate openmmlab

//下载GPU版pytorch(建议pytorch官网查看对应的版本)
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

安装依赖

使用 pip 安装 openmin

使用 min 安装 MMEngine 和 MMCV。

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0rc2"

安装mmdetection

从源代码构建

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .

跑demo验证安装

安装mmrotate

git clone https://github.com/open-mmlab/mmrotate.git -b dev-1.x
cd mmrotate
pip install -v -e .

采用官网提供的验证是否安装成功的方法

（1）验证

mim download mmrotate --config oriented-rcnn-le90_r50_fpn_1x_dota --dest .

（2）推理

python demo/image_demo.py demo/demo.jpg oriented-rcnn-le90_r50_fpn_1x_dota.py oriented_rcnn_r50_fpn_1x_dota_le90-6d2b2ce0.pth --out-file result.jpg

conda list 查看虚拟环境，可以看到需要的包都已安装（openmim、mmcv、mmengine、mmdet、mmrotate）

遇到的问题

powershell无法进入anconda的虚拟环境

想了解计算机上的现用执行策略，打开 PowerShell 然后输入：

>> get-executionpolicy
Restricted

更改执行策略，以管理员身份打开 PowerShell 输入：

>> set-executionpolicy remotesigned

选择“是”，即可。

如果要更改回Windows 客户端计算机的默认执行策略，则设置为restricted：

set-executionpolicy restricted

开启以后poweshell自动会进入base目录

如果不想每次一启动Shell 就自动激活Base 环境

在终端输入 conda config --set auto_activate_base false ，即可。

conda config --set auto_activate_base false

如果又反悔了，想显示了：

conda config --set auto_activate_base true