search

zhixuhao / unet

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

ValueError: Error when checking target: expected conv2d_11 to have shape (16, 16, 1) but got array with shape (256, 256, 1) #175

Open manvirvirk opened 4 years ago

manvirvirk commented 4 years ago

i m training this model on rgb images and grayscale mask. I m getting this error. Can anyone help. Thanks

Nelson-Gon commented 4 years ago

You need to reshape your images to the target size (256,256) or change the target size. You can try to use my fork and see if it works for you.