mediapipe_model_maker.model_util.load_keras_model

Loads a tensorflow Keras model from file and returns the Keras model.

mediapipe_model_maker.model_util.load_keras_model(
    model_path: str, compile_on_load: bool = False
) -> tf.keras.Model

model_path Absolute path to a directory containing model data, such as //saved_model/.
compile_on_load Whether the model should be compiled while loading. If False, the model returned has to be compiled with the appropriate loss function and custom metrics before running for inference on a test dataset.

A tensorflow Keras model.