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
)

Attributes
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
)

Class Variables
alpha	0.75
dropout_rate	0.2