![]() |
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]
|