caffe：5D BLOBがプールされていますか？

Question

私は1x8x128x128のような5D BLOBを持っています。私は5D BLOBを処理できる畳み込みレイヤーを持っています。私はプールレイヤーを使用したいが、動作しない。 と5D BLOBをどうやって使用しますか？

チェックに失敗しました：4 ==底[0] - > num_axes（）（4対5）入力（NUM、チャネル、高さ、幅）

に対応する、4つの 軸を有していなければなりません

私はそれがまだcaffeによってサポートされていないと思う。畳み込みレイヤーを使用して、poolingを使用できますか？ 4Dへ

Answer 1

あなただけの最初の2つの空間次元をプールしたい場合は、次のことができ"Reshape"バック5Dに、（チャネルおよび時間的次元を「退治」）プールし、その後"Reshape"：

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にあります。