Learn the latest in machine learning, generative AI, and more at WiML Symposium 2023Register

hub.get_expected_image_size

Returns expected [height, width] dimensions of an image input.

hub.get_expected_image_size(
    module_or_spec, signature=None, input_name=None
)

Used in the tutorials
Classify Flowers with Transfer Learning

Args
`module_or_spec`	a Module or ModuleSpec that accepts image inputs.
`signature`	a string with the key of the signature in question. If None, the default signature is used.
`input_name`	a string with the input name for images. If None, the conventional input name `images` for the default signature is used.

Returns
A list if integers `[height, width]`.

Raises
`ValueError`	If the size information is missing or malformed.

Args