search

zhixuhao / unet

unet for image segmentation
MIT License
4.6k stars 2k forks source link

Error when checking target: expected conv2d_134 to have 4 dimensions, but got array with shape (24, 320, 320) #195

Open idealove opened 4 years ago

idealove commented 4 years ago

Error when checking target: expected conv2d_134 to have 4 dimensions, but got array with shape (24, 320, 320)

Nelson-Gon commented 4 years ago

Could you add a sample of your code? This is probably to do with using the wrong input shape or using a pretrained model whose weights expect different shapes. Hard to tell without sample code.