mediapipe_model_maker.image_classifier.ModelOptions

Configurable options for image classifier model.

mediapipe_model_maker.image_classifier.ModelOptions(
    dropout_rate: float = 0.2, alpha: float = 0.75
)

dropout_rate The fraction of the input units to drop, used in dropout layer.
alpha Float, larger than zero, controls the width of the network, Only used in mobilenet_v2_keras_spec model spec.

Methods

__eq__

__eq__(
    other
)

alpha 0.75
dropout_rate 0.2