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 = '',
    mean_rgb: Optional[Sequence[float]] = None,
    stddev_rgb: Optional[Sequence[float]] = None
)

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.
mean_rgb	Normalizing mean RGB.
stddev_rgb	Normalizing std RGB.