Module: mediapipe_model_maker.image_classifier

MediaPipe Model Maker Python Public API For Image Classifier.

Classes

class Dataset: Dataset library for image classifier.

class HParams: The hyperparameters for training image classifiers.

class ImageClassifier: ImageClassifier for building image classification model.

class ImageClassifierOptions: Configurable options for building image classifier.

class ModelOptions: Configurable options for image classifier model.

class ModelSpec: Specification of image classifier model.

class SupportedModels: Image classifier model supported by model maker.