如何将.pb文件转换为.h5。 （到Keras的Tensorflow模型）

Question

我已经使用tensorflow对模型进行了重新训练，现在想使用keras来避免会话问题。如何将.pb文件转换为.h5

Answer 1

import tensorflow as tf
from tensorflow.keras.models import save_model, Sequential

model_path = r"c:\temp\model.pb"

model = tf.keras.models.load_model(model_path)

save_model(model,model_path + r"\new_model.h5", save_format='h5')

Answer 2

我发现此链接很有用：https://backstreetcoder.com/convert-tensorflow-pb-model-to-keras-h5-model/

显然，在最初加载模型时，您需要使用封闭目录名称而不是 .pb 文件名。这解决了错误：

SavedModel file does not exist at: saved_model_dir/{saved_model.pbtxt|saved_model.pb}