Tensorflow与Caffe的反卷积

时间:2017-12-16 10:24:38

标签: tensorflow deep-learning caffe deconvolution

Caffe和Tensorflow中的Deconvolution层有什么区别?在Tensoroflow中,有两种填充模式:" SAME"和" VALID",哪一个等于Caffe中使用的填充模式?

例如,在Tensorflow中,Deconvolution图层是:

decv = slim.convolution2d_transpose(in_layer, num_outputs=256, kernel_size=[8, 8], stride=4, padding='VALID', scope='decv')

在Caffe中,它是:

layer { 
  name: "decv"
  type: "Deconvolution"
  bottom: "some_layer"
  top: "some_layer"
  param { lr_mult: 1 decay_mult: 1.0 }
  param { lr_mult: 2 decay_mult: 0}
  convolution_param {
    num_output: 256
    pad: 1 stride: 4 kernel_size: 8
    group: 256 
    weight_filler { type: "bilinear" }
  }
}

将代码从Caffe传输到Tensorflow时,我无法实现相同的上采样映射。

1 个答案:

答案 0 :(得分:0)

'VALID'表示没有填充,但你的Caffe图层用pad填充:1。

尝试'相同'填充