caffe使用FCN时,训练loss总是保持不变的问题

使用原作者分享出来的FCN版本进行训练,如果直接使用.prototxt文件在caffe上训练,会出现loss基本保持不变的问题。

这种问题的产生是因为原作者搭建网络时设置的问题,即初始化的问题:

layer {
  name: "upscore_pool4"
  type: "Deconvolution"
  bottom: "fuse_pool4"
  top: "upscore_pool4"
  param {
    lr_mult: 0
  }
  convolution_param {
    num_output: 21
    bias_term: false
    kernel_size: 4
    stride: 2
  }
}

在定义Deconvolution层的时候

 param {
    lr_mult: 0  #不更新卷积核

  }

bias_term: false #不添加偏置

 convolution_param {
    num_output: 21
    bias_term: false
    kernel_size: 4
    stride: 2

  } #没有指定卷积核的初始化方式,即所有权值为默认值0

这种定义Deconvolution的方式会导致所有的输出feature map均为0.

所以如果是二分类问题,则



评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值