Ubuntu20运行SegNeXt代码提取道路水体(六)——使用mmsegmentation里面的预设网络跑自己的数据集及调试网络

configs文件夹里面有许多其他网络
都是已经配置好的
我们可以按照上一篇博文里面的步骤应用到我们自己的数据集上跑
需要修改的内容有：
1、models里面的网络配置

norm_cfg = dict(type='BN', requires_grad=True)

test_cfg=dict(mode='slide', crop_size=(512, 512), stride=(256, 256)))

num_classes=2#修改成自己的类别数量

2、数据集datasets里面的设置
为了我们的效果能更好些
对预测的图片进行一个配置
设置一个比率

img_ratios=[0.5, 0.75, 1.0, 1.25, 1.5, 1.75, 2.0]