It seems your model expect an image to have 3 channels but you are feeding an image with one channel. The printed shapes are different. 5 of them have 3 channel and one has 1 channel. So the network should work for first 5 images and not for the 6th image, right?