caffe:转换层滤波器的平均值(通道)

时间:2018-06-18 08:19:25

标签: neural-network deep-learning caffe conv-neural-network pycaffe

我有一个维度nXmx16x1的转换层,我想计算16个滤波器的平均值或总和,因此结果是1个暗淡的nxmx1x1滤波器。

类似于matlab中的均值(A,3)

1 个答案:

答案 0 :(得分:1)

使用常量1x1 "Conv"过滤器:

layer {
  name: "avg"
  type: "Convolution"
  bottom: "input"
  top: "output"
  param { lr_mult: 0 decay_mult: 0 }  # fix this layer, do not train
  convolution_param {
    num_output: 1
    kernel_size: 1
    bias: false  # no bias
    weight_filler: { type: "constant" value: 0.0625 }  # 1/16.
  }
}