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.