View source on GitHub |
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.