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

tflite_model_maker.image_classifier.EfficientNetLite0Spec

Creates EfficientNet-Lite0 model spec. See also: tflite_model_maker.image_classifier.ModelSpec.

tflite_model_maker.image_classifier.EfficientNetLite0Spec(
    *,
    uri='https://hub.tensorflow.google.cn/tensorflow/efficientnet/lite0/feature-vector/2',
    compat_tf_versions=[1, 2],
    input_image_shape=None,
    name='efficientnet_lite0'
)

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.