一、源码：

https://github.com/sithu31296/semantic-segmentation

我并没使用SegFormer的官方源码，那个mmcv特磨人了，各种奇葩配置错误。

二、环境配置

1. 新建conda环境

conda create -n segformer3715 python=3.7.15

选用python3.7.15(纯粹的3.7.0版本，竟然搞出了cannot import name 'OrderedDict' from 'typing错误，不太好改）

2. 安装pytorch

用的轮子离线安装

pip install torch-1.13.0+cu117-cp37-cp37m-win_amd64.whl

和

pip install torchvision-0.14.0+cu117-cp37-cp37m-win_amd64.whl

三、infer试一下

1、目标

python tools/infer.py --cfg configs/ade20k.yaml

2、下载文件

按照ade20k.yaml，下载相应文件。

这些文件的链接，都在README.md文件中

3、重要提示：pycharm的终端，一定要选择 command prompt

pycharm的终端，一定要选择 command prompt，否则输入指令没反应。

四、train试一下

各种问题来了：

需要像我一样改一下ade20k.yaml

同时，我担心我还是用错了cuda，我在train文件中，改了这行代码如下：