Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

tflite_model_maker.image_classifier.MobileNetV2Spec

Creates MobileNet v2 model spec. See also: tflite_model_maker.image_classifier.ModelSpec.

tflite_model_maker.image_classifier.MobileNetV2Spec(
    *,
    uri='https://hub.tensorflow.google.cn/google/tf2-preview/mobilenet_v2/feature_vector/4',
    compat_tf_versions=2,
    input_image_shape=None,
    name='mobilenet_v2'
)

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