caffe：5D斑点池？

Question

我有一个5D blob像1x8x128x128，我有一个能够处理我的5D blob的卷积层。当我想使用一个池层，尽管它不起作用。你如何使用带有5D blob的pool-layer？

检查失败：4 ==底部[0] - > num_axes（）（4与5）输入必须有4个 轴，对应于（NUM，信道，高度，宽度）

我认为它只是不被caffe支持。我可以只使用卷积层并执行pooling？

Answer 1

如果你想集中只有前两个空间维度，可以"Reshape"到4D（“挤压”的渠道和时间维度），池，然后"Reshape"回5D：

layer { 
    name: "pool/reshape4D" 
    type: "Reshape" 
    bottom: "in" 
    top: "pool/reshape4D" 
    reshape_param { axis: 1 num_axes: 1 shape { dim: -1 } } 
} 
layer { 
    name: "pool" 
    type: "Pooling" 
    bottom: "pool/reshape4D" 
    top: "pool" 
    # pooling params here... 
} 
layer { 
    name: "pool/reshape5D" 
    type: "Reshape" 
    bottom: "pool" 
    top: "pool/reshape5D" 
    reshape_param { axis: 1 num_axes: 1 shape { dim: -1 dim: <temporal_dim> } } # replace <.> with the actual temporal dimension size. 
} 

见定义ReshapeParameter在caffe.proto有关更多详情。