tkuanlun350 / 3DUnet-Tensorflow-Brats18

3D Unet biomedical segmentation model powered by tensorpack with fast io speed
202 stars 68 forks source link

"Dimension 0 in both shapes must be equal,but are 4 and 3" #37

Open yan4821567 opened 4 years ago

yan4821567 commented 4 years ago

I run the code with patch '128x128x128', and cuda out of memory. So I change patch size to '20x128x128',and it says that:

ValueError: Dimension 0 in both shapes must be equal, but are 4 and 3. Shapes are [4,16,16] and [3,16,16]. for 'tower0/unet3d/concat' (op: 'ConcatV2') with input shapes: [1,128,4,16,16], [1,128,3,16,16], [] and with computed input tensors: input[2] = <1>.

But when I change patch size to '64x64x64', it works. But I am not sure if this is the best setting.