hub.get_expected_image_size

View source on GitHub

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

Used in the notebooks

Used in the tutorials

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.

A list if integers [height, width].

ValueError If the size information is missing or malformed.