Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

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 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.