Ubuntu20运行SegNeXt代码提取道路水体(一)——从零开始运行代码过程摸索

SegNeXt代码最近可谓是非常火

应导师的要求打工人需要学习一下新的代码

但是网上千篇一律都是论文的讲解
如何跑通代码并且使用自己的数据跑出一个好的结果却没有一个详细的教程

那么就让我自己来从零开始跑通代码吧

下载代码和数据

首先咱们先别想着用自己的数据 从github上下载后发现代码量还是巨大的
看得我头晕眼花
也不知道数据的格式
那我们先去下载该论文使用的数据来跑通吧
首先下载代码
然后简介里面有这个

跳转到指定的链接MMSegmentation
跳转后发现
Windows is in experimental support
所以这里使用Ubuntu系统来运行该程序

这里下载v0.24.1版本的
然后按照他的说法进行安装
首先安装mmcv

注意避雷
下面的语句错误：

---

我使用了下面的语句

然后可以看到自己的下载的版本

然后再使用这个语句在下载的0.24.1目录下进行安装

python setup.py develop
1

但这里报错！！！！！注意！！！！
不能下载1.7版本的mmcv

---

于是我重新下载了1.6.0版本的
输入下面的语句

mim install mmcv-full==1.6.0
1

就成功啦
但是包之间的安装依赖有一些问题
需要不断调整
这俩个包竟然产生了冲突


这么看来就需要把一个包的版本调低
我调整了
先卸载这个高版本的

再安装

最后再安装共有的这个包

最后demo文件报错

问题解决：SyntaxError: future feature annotations is not defined

这个需要看上面报错的包
我是这个
File “/home/wangtianni/.conda/envs/pytorch/lib/python3.6/site-packages/prettytable-3.5.0-py3.6.egg/prettytable/init.py”, line 1
from future import annotations

参考这篇文章
同理 出现问题的应该是prettytable
于是安装低一点的包
这样就不报错啦！

运行实例文件

参考这篇文章
但是死活都没有这个pth文件呀

找了github上最新的安装包也没有这个checkpoints的包
但最后我终于在config目录下的对应的py文件目录中找到了它的下载地址！！！
自己新建了一个checkpoints文件夹把对应的pth文件放进去
用这个语句

python demo/image_demo.py demo/demo.png configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py checkpoints/pspnet_r50b-d8_512x1024_80k_cityscapes_20201225_094315-6344287a.pth --device cuda:0 --palette cityscapes
1

运行出来的结果

效果并不好 而且还警告文件对应不上
应该是py文件和pth文件对应不上
于是我自己修改了一下语句

python demo/image_demo.py demo/demo.png configs/pspnet/pspnet_r50b-d8_512x1024_80k_cityscapes.py checkpoints/pspnet_r50b-d8_512x1024_80k_cityscapes_20201225_094315-6344287a.pth --device cuda:0 --palette cityscapes
1

这样效果要好些了！

但如何使用自己的数据来训练推理还有些复杂
可以参考这篇文章进行学习