Module: tflite_model_maker.image_classifier

APIs to train an image classification model.

Task guide:

https://www.tensorflow.org/lite/tutorials/model_maker_image_classification

Classes

class DataLoader: DataLoader for image classifier.

class ImageClassifier: ImageClassifier class for inference and exporting to tflite.

class ModelSpec: A specification of image model.

Functions

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

EfficientNetLite1Spec(...): Creates EfficientNet-Lite1 model spec. See also: tflite_model_maker.image_classifier.ModelSpec.

EfficientNetLite2Spec(...): Creates EfficientNet-Lite2 model spec. See also: tflite_model_maker.image_classifier.ModelSpec.

EfficientNetLite3Spec(...): Creates EfficientNet-Lite3 model spec. See also: tflite_model_maker.image_classifier.ModelSpec.

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

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

Resnet50Spec(...): Creates ResNet 50 model spec. See also: tflite_model_maker.image_classifier.ModelSpec.

create(...): Loads data and retrains the model based on data for image classification.