Attention: This MediaPipe Solutions Preview is an early release. Learn more.

mediapipe_model_maker.image_classifier.ModelSpec

Stay organized with collections Save and categorize content based on your preferences.

Specification of image classifier model.

uri str, URI to the pretrained model.
input_image_shape list of int, input image shape. Default: [224, 224].
name str, model spec name.

mean_rgb [0.0]
stddev_rgb [255.0]