Introducing LiteRT : Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite.
Learn more
Send feedback
mediapipe_model_maker.image_classifier.SupportedModels
Stay organized with collections
Save and categorize content based on your preferences.
Image classifier model supported by model maker.
Class Variables
EFFICIENTNET_LITE0
<SupportedModels.EFFICIENTNET_LITE0: functools.partial(<class 'mediapipe_model_maker.python.vision.image_classifier.model_spec.ModelSpec'>, uri='https://tfhub.dev/tensorflow/efficientnet/lite0/feature-vector/2', name='efficientnet_lite0')>
EFFICIENTNET_LITE2
<SupportedModels.EFFICIENTNET_LITE2: functools.partial(<class 'mediapipe_model_maker.python.vision.image_classifier.model_spec.ModelSpec'>, uri='https://tfhub.dev/tensorflow/efficientnet/lite2/feature-vector/2', input_image_shape=[260, 260], name='efficientnet_lite2')>
EFFICIENTNET_LITE4
<SupportedModels.EFFICIENTNET_LITE4: functools.partial(<class 'mediapipe_model_maker.python.vision.image_classifier.model_spec.ModelSpec'>, uri='https://tfhub.dev/tensorflow/efficientnet/lite4/feature-vector/2', input_image_shape=[300, 300], name='efficientnet_lite4')>
MOBILENET_V2
<SupportedModels.MOBILENET_V2: functools.partial(<class 'mediapipe_model_maker.python.vision.image_classifier.model_spec.ModelSpec'>, uri='https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4', name='mobilenet_v2')>
MOBILENET_V2_KERAS
<SupportedModels.MOBILENET_V2_KERAS: functools.partial(<class 'mediapipe_model_maker.python.vision.image_classifier.model_spec.ModelSpec'>, uri=None, name='mobilenet_v2_keras', mean_rgb=[127.5], stddev_rgb=[128.0])>
Send feedback
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License , and code samples are licensed under the Apache 2.0 License . For details, see the Google Developers Site Policies . Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-05-07 UTC.
Need to tell us more?
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-05-07 UTC."],[],[]]
View source on GitHub