hub.get_num_image_channels

Returns expected num_channels dimensions of an image input.

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

This is for advanced users only who expect to handle modules with image inputs that might not have the 3 usual RGB channels.

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
An integer with the number of input channels to the module.

Raises
ValueError	If the channel information is missing or malformed.