mediapipe_model_maker.image_classifier.ImageClassifierOptions

Configurable options for building image classifier.

mediapipe_model_maker.image_classifier.ImageClassifierOptions(
    supported_model: mediapipe_model_maker.image_classifier.SupportedModels,
    model_options: Optional[mediapipe_model_maker.image_classifier.ModelOptions] = None,
    hparams: Optional[mediapipe_model_maker.image_classifier.HParams] = None
)

supported_model A model from the SupportedModels enum.
model_options A set of options for configuring the selected model.
hparams A set of hyperparameters used to train the image classifier.

Methods

__eq__

__eq__(
    other
)

hparams None
model_options None