Attention: This MediaPipe Solutions Preview is an early release. Learn more.

Module: mediapipe_model_maker.image_classifier.image_classifier

Stay organized with collections Save and categorize content based on your preferences.

APIs to train image classifier model.


classification_ds module: Common classification dataset library.

classifier module: Custom classifier.

hp module: Hyperparameters for training image classification models.

image_classifier_options module: Options for building image classifier.

image_preprocessing module: ImageNet preprocessing.

model_opt module: Configurable model options for image classifier models.

model_util module: Utilities for models.

ms module: Image classifier model specification.

quantization module: Libraries for post-training quantization.

train_image_classifier_lib module: Library to train model.


class ImageClassifier: ImageClassifier for building image classification model.