Win10下 Swin Transformer目标检测环境配置流程

Win10下 Swin-Transformer目标检测环境；对于第一次配置mmdetection的同学可能有所帮助。

1.版本环境介绍

Window 10
Visual Studio 2017
python3.7
cuda 10.0
cudnn 7.6.5
pytorch 1.8.0
torchvision 0.9.0
mmcv-full 1.3.8
项目源码下mmdet 2.11.0

2.mmcv-full

• 这里需要先安装好pytorch 1.8与vs 2017，并将vs 2017的cl.exe添加环变量；

添加下述文件夹至环境变量；由于安装路径不同，个人路径不同。
#####\Microsoft Visual Studio\2017\Community\VC\Tools\MSVC\14.####\bin\Hostx86\x64

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

#build
python setup.py build_ext
# install
python setup.py develop

• 最后pip list 查看mmcv版本，当前是版本：1.3.8 。

3.项目下的mmdet安装

• 切换到Swin-Transformer-Object-Detection项目下，项目代码包含了mmdet库，然后编译安装mmdet。

cd Swin-Transformer-Object-Detection-master
#安装所有依赖
pip install -r requirements.txt 
#编译安装mmdet
python setup.py develop

• 最后pip list 查看mmdet版本，当前是版本：2.11.0 。

4.apex安装

git clone https://github.com/NVIDIA/apex
cd apex-master
#编译安装
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .

• 最后pip list 查看apex版本，当前是版本：0.1 。

5.demo

• 首先下载官网预训练模型及对应config；

这里我下载了 :Swin-T (mask_rcnn)
mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py
mask_rcnn_swin_tiny_patch4_window7_1x.pth

#单图像推理
python demo/image_demo.py demo/demo.jpg configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py ./checkpoints/mask_rcnn_swin_tiny_patch4_window7_1x.pth
#摄像头推理
python demo/webcam_demo.py configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py ./checkpoints/mask_rcnn_swin_tiny_patch4_window7_1x.pth
#视频推理并显示
python demo/video_demo.py ./demo/demo.mp4 configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py ./checkpoints/mask_rcnn_swin_tiny_patch4_window7_1x.pth --show

6.参考地址

原文地址：https://arxiv.org/abs/2103.14030
项目地址：https://github.com/SwinTransformer/Swin-Transformer-Object-Detection
mmcv文档：https://mmcv.readthedocs.io/en/latest/build.html
apex：https://github.com/NVIDIA/apex