如何将张量流模型转换为keras模型？ .pb文件到.hdf5？

Question

我在Google上搜索了几个小时，但没有看到要转换的任何代码/包。
任何对此的指导将不胜感激。

Answer 1

首先加载.pb模型，然后另存为keras .h5模型。

best_model_file_pb = MODEL_DIR + "best_model.pb"
best_model_file_h5 = MODEL_DIR + "best_model_.h5"
loaded_model = tf.keras.models.load_model(best_model_file_pb)
tf.keras.models.save_keras_model(loaded_model, best_model_file_h5)
loaded_model_from_h5 = tf.keras.models.load_model(best_model_file_h5)

此转换后的模型将使用tf.keras在tensorflow生态系统中工作。但是，如果您希望tf.keras模型在keras（带有tensorflow后端）中运行，则该模型将不起作用。

Answer 2

使用tf.keras.models.load_model('Model')

代替tf.keras.models.load_model('saved_model.pb')

检查类似问题 https://stackoverflow.com/a/62211811/14290244 很好地解释了 .pb 到 .H5 格式的转换。