hub.get_expected_image_size

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

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

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.