MobileNetV5ImageClassifierPreprocessor layer

[source]

MobileNetV5ImageClassifierPreprocessor class

keras_hub.models.MobileNetV5ImageClassifierPreprocessor(
    image_converter=None, **kwargs
)

Base class for image classification preprocessing layers.

ImageClassifierPreprocessor tasks wraps a keras_hub.layers.ImageConverter to create a preprocessing layer for image classification tasks. It is intended to be paired with a keras_hub.models.ImageClassifier task.

All ImageClassifierPreprocessor take inputs three inputs, x, y, and sample_weight. x, the first input, should always be included. It can be a image or batch of images. See examples below. y and sample_weight are optional inputs that will be passed through unaltered. Usually, y will be the classification label, and sample_weight will not be provided.

The layer will output either x, an (x, y) tuple if labels were provided, or an (x, y, sample_weight) tuple if labels and sample weight were provided. x will be the input images after all model preprocessing has been applied.

All ImageClassifierPreprocessor tasks include a from_preset() constructor which can be used to load a pre-trained config and vocabularies. You can call the from_preset() constructor directly on this base class, in which case the correct class for your model will be automatically instantiated.

Examples.

preprocessor = keras_hub.models.ImageClassifierPreprocessor.from_preset(
    "resnet_50",
)

# Resize a single image for resnet 50.
x = np.random.randint(0, 256, (512, 512, 3))
x = preprocessor(x)

# Resize a labeled image.
x, y = np.random.randint(0, 256, (512, 512, 3)), 1
x, y = preprocessor(x, y)

# Resize a batch of labeled images.
x, y = [
    np.random.randint(0, 256, (512, 512, 3)),
    np.zeros((512, 512, 3))
], [1, 0]
x, y = preprocessor(x, y)

# Use a [`tf.data.Dataset`](https://www.tensorflow.org/api_docs/python/tf/data/Dataset).
ds = tf.data.Dataset.from_tensor_slices((x, y)).batch(2)
ds = ds.map(preprocessor, num_parallel_calls=tf.data.AUTOTUNE)

image_converter property

keras_hub.models.MobileNetV5ImageClassifierPreprocessor.image_converter

The image converter used to preprocess image data.