Windows下配置MMDetection

最新推荐文章于 2024-04-25 21:20:06 发布

原创

最新推荐文章于 2024-04-25 21:20:06 发布 · 1k 阅读

2 ·

CC 4.0 BY-SA版权

本文介绍在Windows下配置MMDetection的方法。因官方最新版侧重Linux环境，故采用较早版本。配置内容包括安装pytorch 1.6.0+cu101、torchvision 0.6.1、mmcv - full 1.1.5和MMDetection2.6，还涉及编译源码、检查正确性及案例应用等步骤。

文章目录

Windows配置MMDetection说明
安装mmcv-full和Pytorch
安装MMDetection2.6
编译源码
检查正确性
案例应用

Windows配置MMDetection说明

由于官方最新版本侧重linux环境配置，windows目前更新并不及时，本文采用比较早的版本作为windows下的环境的配置。
配置安装内容如下：

pytorch:1.6.0+cu101
torchvision:0.6.1
mmcv-full==1.1.5
mmdetection2.6

安装mmcv-full和Pytorch

mmcv-full包括cuda套件部分，可以用于GPU加速，

conda create -n mmd26 python=3.7
conda activate mmd26

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.6.1 -f https://download.pytorch.org/whl/torch_stable.html


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