input_img = Input((im_height, im_width, 1), name='img') code example

Example: image size

<input id="image" type="image" width="100" height="30" alt="Login"
       src="https://raw.githubusercontent.com/mdn/learning-area/master/html/forms/image-type-example/login.png">