Module: tflite_model_maker.model_spec

APIs for the model spec of TFLite Model Maker.

Functions

get(...): Gets model spec by name or instance, and init with args and kwarges.

AUDIO_CLASSIFICATION_MODELS ['audio_browser_fft', 'audio_teachable_machine', 'audio_yamnet']
IMAGE_CLASSIFICATION_MODELS ['efficientnet_lite0', 'efficientnet_lite1', 'efficientnet_lite2', 'efficientnet_lite3', 'efficientnet_lite4', 'mobilenet_v2', 'resnet_50']
OBJECT_DETECTION_MODELS ['efficientdet_lite0', 'efficientdet_lite1', 'efficientdet_lite2', 'efficientdet_lite3', 'efficientdet_lite4']
QUESTION_ANSWER_MODELS ['bert_qa', 'mobilebert_qa', 'mobilebert_qa_squad']
RECOMMENDATION_MODELS ['recommendation']
TEXT_CLASSIFICATION_MODELS ['bert_classifier', 'average_word_vec', 'mobilebert_classifier']