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
)

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.