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

tflite_model_maker.image_classifier.EfficientNetLite4Spec

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

tflite_model_maker.image_classifier.EfficientNetLite4Spec(
    *,
    uri='https://hub.tensorflow.google.cn/tensorflow/efficientnet/lite4/feature-vector/2',
    compat_tf_versions=[1, 2],
    input_image_shape=[300, 300],
    name='efficientnet_lite4'
)

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.