将 tensorflow 保存的模型加载为 keras 模型

Loading a tensorflow saved model as a keras model

提问人:Dani 提问时间:11/15/2023 最后编辑:Dani 更新时间:11/15/2023 访问量:17

问:

我有这个用于动作识别的 MoViNet a0 模型。我想将模型 (.pb) 加载为 Keras 模型。我的目标是:修改模型,调整图层等...,使用Keras高级API。

模型文件如下:

  • variables 目录

  • keras_metadata.pb

  • saved_model.pb

我尝试用这个加载模型:

loaded_model = tf.saved_model.load(saved_model_path)

saved_model_path包含模型目录的路径。

但是,类型为“_UserObject”的最终对象不提供与 Keras 模型相同的灵活性。

我试图使用这个来获取签名:

print(loaded_model.signatures)

获取以下签名密钥:“call”“init_states”

我试过这个:

cloned_model = tf.keras.models.clone_model(loaded_model)

出现以下错误:类型对象“_UserObject”没有属性“from_config”

是否可以将模型加载为 keras 模型?

python-3.x keras tensorflow2.0

评论


答: 暂无答案