MMSegmentation遇到的问题汇总(不断更新)

首先说明一下我的运行环境是在win10下的

遇到的第一个问题就是在训练的时候不知道在哪里能运行模型,后来参考mmsegmentation文档才明白该在哪里进行train,具体的train文件是在tools下面的train文件

为了方便起见我并没有在命令行配置参数,而是直接修改train文件

config我默认的default是我自己根据数据集对原模型进行修改而得到的,当然也可以直接用config里面的模型文件,workdir是输出的log文件路径

同时数据集文件里面我用的直接路径,并没有进行连接,也一样可以对数据集进行训练,而另外一些其他的参数需要在你选择使用的模型文件进行修改了,这里我们以fcn为例。

 

 我们可以看到fcn里面有很多模型,r50指的是backbone使用resnet50模型,80k指的是迭代次数,在mmseg里面没有用epoch而是用的iter,区别在于iter是一个batchsize输入到网络的次数,而epoche是所有的数据遍历的次数。  

首先第一个normcfg,因为我是单卡训练,因此改成了BN,下面主要有三个结构,分别是主干网络backbone,主解码头decode_head以及辅助解码头aux_head,numstage是主干网络生成特征图的数目,dilation为空洞率,其他都是一些比较基础甚至不太需要变的参数,loss_decode这里有两个,分别是辅助解码以及主解码的损失函数,主解码占的比重比较大,而辅助损失比较小,这里如果想修改损失函数的话,可以采用如下修改方式

loss_decode=[dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0,class_weight=[0.001,1]),
            dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0,class_weight=[0.001,1])]

而如果想每轮都进行验证集评价,需要在workflow里面加入val

 

同时在base文件里面

对于val_step函数做出如下修改,就可以显示验证集的数据了

 

 

  • 5
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值