将与tf.saved_model.save一起保存的SavedModel格式的Tensorflow模型(.pb文件)转换



我正在尝试将以SavedModel格式保存的Tensorflow对象检测模型(在包含.pb文件、assets文件夹和variables文件夹的文件夹中(转换为Keras.h5,以进行更简单的推断。然而,尝试常见的StackOverflow对这个问题的回答(例如这个(会导致ValueError: Unable to create a Keras model from this SavedModel. This SavedModel was created with tf.saved_model.save, and lacks the Keras metadata.Please save your Keras model by calling model.save or tf.keras.models.save_model.

使用Tensorflow对象检测API对模型进行训练并保存。是否仍然可以将其转换为Keras模型?如果是,有什么建议吗?

我在将YoloV5转换为Tensorflow格式以部署模型时遇到了同样的问题。您需要在模型的导出命令中使用keras=True。这将在目录中生成Keras_metadata.pb文件。请查看此链接。

最新更新