[nerdponx]

I haven't even looked at the library, but wouldn't input_shape just be the shape of the image, in pixels?

[reachtarunhere]

Yes that would be it. Along with Channels and number of images you are processing at a time (batch).