caffe : 5D blobs pooling?

Question

나는 1x8x128x128과 같은 5D BLOB를 가지고 있으며, 5D BLOB를 처리 할 수있는 Convolution 레이어가 있습니다. 풀 레이어를 사용하고 싶지만 작동하지 않습니다. 을 5D BLOB와 어떻게 사용합니까?

확인에 실패 4 == 하단 [0] -> num_axes() (4 대 5) 입력 (NUM, 채널 높이, 폭)

에 대응하는 4 축 있어야

나는 그것이 아직 caffe에 의해 지원되지 않는다고 생각한다. 난 그냥 convolution 레이어를 사용하여 pooling 할 수 있을까요?

Answer 1

만 처음 두 개의 공간 차원을 풀로 할 경우, 수, 수영장 및 5D 뒤쪽 "Reshape" (채널 및 시간 기준 "부수") 내지 4D "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에 입력하십시오.