caffe CNN:跨渠道汇聚

时间:2018-06-18 16:22:21

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

我有一个nxmx16x1转换层,我想在整个通道中进行合并,因此结果的维度为nxmx1x1。

有什么建议吗?据我所知,汇集没有轴参数,对吗?

1 个答案:

答案 0 :(得分:1)

你不需要axis param。只需使用非均匀内核/步幅:

layer {
  name: "pool16x1"
  type: "Pool"
  bottom: "input"
  top: "output"
  pooling_param {
    kernel_size: 16
    kernel_size: 1
    stride: 16
    stride: 1
    # ...
   }
}

应该做的伎俩...