caffe - チェックに失敗しました: top_shape[j] == bottom[i]->shape(j): 3D データの Concat レイヤー用

Question

で3D-Unetをトレーニングしようとしていcaffeます。異なるボリュームの幅、高さ、深さは異なります。HDF5 データセットの最初のボリュームの入力形状は1 1 104 281 389(NxCxDxHxW) で、Concat layer(concat_d2c_u2a-b) に達するとエラーが発生します。入力は次の 2 つの形状です。

• 1 256 19 64 91 scaled2c_relu_d2c_0_split_1
• 1 512 12 56 84 スケールu2a

この問題を解決するにはどうすればよいですか? 次元の順番のせいですか？

I0708 22:19:59.811188 30282 layer_factory.cpp:74] Creating layer concat_d2c_u2a-b
I0708 22:19:59.811198 30282 net.cpp:185] Creating Layer concat_d2c_u2a-b
I0708 22:19:59.811200 30282 net.cpp:549] concat_d2c_u2a-b <- scaleu2a
I0708 22:19:59.811203 30282 net.cpp:549] concat_d2c_u2a-b <- scaled2c_relu_d2c_0_split_1
I0708 22:19:59.811208 30282 net.cpp:515] concat_d2c_u2a-b -> u2b
F0708 22:19:59.811223 30282 concat_layer.cpp:45] Check failed: top_shape[j] == bottom[i]->shape(j) (12 vs. 19) All inputs must have the same shape, except at concat_axis.
*** Check failure stack trace: ***