【语义分割】Swin-Transformer-Semantic-Segmentation 代码windows环境配置

1、Anaconda虚拟环境创建、激活、删除等

打开Anaconda Powershell Prompt

 conda create -n swin_test python=3.9//创建一个新的虚拟环境
 conda activate swin_test //进入刚创建的虚拟环境

//若环境配置错误可以全部删除再来
 conda remove -n swin_test --all //删除虚拟环境

2、torch、torchvision、cuda版本对应及安装

这三个安装需注意版本对应，也需要考虑到mmcv的版本是否支持。

我用的cuda版本为10.2，torch版本为1.8.1，torchvision版本为0.9.1。
三个版本的对应关系可以从https://download.pytorch.org/whl/torch_stable.html查看。
conda10.2可以从官网下载，地址为https://developer.nvidia.com/cuda-toolkit-archive

//在上述虚拟环境下使用如下指令进行安装
pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/
//安装完成后可输入指令查看版本是否正确
pip list

版本list如下，符合要求

接下来开始安装mmcv。

3、mmcv指定版本下载及安装

windows系统下直接pip的mmcv-full的版本最低为1.4.0，起初直接使用了此版本，安装结束运行程序时一直报dll丢失的错误，折腾了好久选择下载mmcv更低版本，自行编译。
自编译需要vs的环境，可以下载vs19，并将其添加入环境变量。本电脑上之前有配置过，所以没有再重新配置，若编译有问题可以查看博客https://blog.csdn.net/Xiao_Bai_hh/article/details/127107411
mmcv指定版本下载：
github上找到官方程序，在master->Tags中找到想要的版本，我试成功用的是1.3.0版本。

下载到本地，将文件夹放到虚拟环境目录下，如下图所示。

然后回到Anaconda Powershell Prompt

cd 刚刚mmcv目录位置 
//环境配置
$env:TORCH_CUDA_ARCH_LIST="6.1" //这个数值根据自己电脑显卡设置，算力查看[*https://developer.nvidia.com/zh-cn/cuda-gpus*](https://developer.nvidia.com/zh-cn/cuda-gpus)
$env:MMCV_WITH_OPS = 1
$env:MAX_JOBS = 8 //根据自己电脑cpu核数配置，不大于电脑cpu核数

官网算力值如下

环境参数设置好好后开始安装mmcv

python setup.py build_ext 

编译需要一定时间，耐心等待，成功的话是这个样子

如果出现报错，可能是vs编译没配置好，请再尝试重新设置下。
同样的再运行

python setup.py develop

成功后pip list查看已安装的包，发现成功安装mmvc，到这已经基本成功了

4、mmdet与mmsegmentation版本对应

mmdet版本为2.11.0，mmseg版本为0.11.0
直接pip就行

pip install mmdet==2.11.0
pip install mmsegmentation==0.11.0

到这环境的版本配置都弄好了

5、执行demo程序

从https://github.com/SwinTransformer/Swin-Transformer-Semantic-Segmentation下载model文件。

然后在终端中运行指令

python demo/image_demo.py demo/demo.png configs/swin/upernet_swin_tiny_patch4_window7_512x512_160k_ade20k.py  upernet_swin_tiny_patch4_window7_512x512.pth

会出现错误

虚拟环境中依次输入

pip install -r requirements.txt
pip install -v -e .
python setup.py develop

运行成功

6、若依旧有问题，这些信息也许能帮助到你

以下的视频和博客帮助我解决了这个问题，感谢！！
swin-transformer环境安装视频
mmcv官方文档
安装mmcv、mmdet等版本问题