INPUT_IMAGE_SIZE = 224