shiba24 / 3d-unet

Chainer implementations of 3D UNet
182 stars 55 forks source link

Regarding Input Shape for Conv3d #1

Open isn4 opened 7 years ago

isn4 commented 7 years ago

Hi, Curious about how you've modified either the layers from tf_unet or modified the input images. tf.nn.conv3d (which i assume is what you substituted tf.nn.conv2d in layers.py for) only takes 5D tensors but all of the tensors I'm seeing in your code are 4D (batch, height, width, channel).