search

MIC-DKFZ / BraTS2017

Apache License 2.0
74 stars 16 forks source link

How can it be transformed into dimension 5 in the line 139:"l = DimshuffleLayer(l, (0, 2, 3, 4, 1))" which seems to increace dimension batch #17

Closed yianzhongguo closed 5 years ago

yianzhongguo commented 5 years ago

@FabianIsensee Hi, sir. According to the line 130 :"l_pred = Conv3DLayer(l, num_output_classes, 1, pad='same', nonlinearity=None)" in the network_architecture.py, the output shape of the network should be 4128128*128 which is dimension 4. How can it be transformed into dimension 5 in the line 139:"l = DimshuffleLayer(l, (0, 2, 3, 4, 1))" which seems to increace dimension batch. Thank you!

FabianIsensee commented 5 years ago

Network output is always 5D

batch, channel, X, Y, Z

yianzhongguo commented 5 years ago

But it seems that the output of l_pred = Conv3DLayer(l, num_output_classes, 1, pad='same', nonlinearity=None) is 4D.

FabianIsensee commented 5 years ago

it's 5D. Every conv3D layer has 5d inputs and outputs

yianzhongguo commented 5 years ago

I know the input of conv3D layer is 5D, but what does the shape of outputs is determined by?

FabianIsensee commented 5 years ago

by the batch size of the input, num_output_classes, padding, stride and input shape

the output here is is 5D with the same batch size and 3d shape of the input and num_output_classes channels

yianzhongguo commented 5 years ago

I've made it clear, thank you very much!