mediapipe_model_maker.model_util.load_keras_model

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

View aliases

Main aliases

mediapipe_model_maker.face_stylizer.face_stylizer.face_stylizer_options.model_opt.loss_functions.model_util.load_keras_model, mediapipe_model_maker.face_stylizer.face_stylizer.loss_functions.model_util.load_keras_model, mediapipe_model_maker.face_stylizer.face_stylizer.model_opt.loss_functions.model_util.load_keras_model, mediapipe_model_maker.face_stylizer.face_stylizer.model_util.load_keras_model, mediapipe_model_maker.face_stylizer.face_stylizer_options.model_opt.loss_functions.model_util.load_keras_model, mediapipe_model_maker.face_stylizer.model_options.loss_functions.model_util.load_keras_model

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

Args
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.

Returns
A tensorflow Keras model.