mediapipe_model_maker.image_classifier.ModelSpec

Specification of image classifier model.

mediapipe_model_maker.image_classifier.ModelSpec(
    uri: str,
    input_image_shape: Optional[List[int]] = None,
    name: str = ''
)

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

Class Variables
mean_rgb	[0.0]
stddev_rgb	[255.0]