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

Question

I've retrained my model using tensorflow and now want to use keras to avoid session stuffs. How can I convert .pb file to .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

I found this link useful: https://backstreetcoder.com/convert-tensorflow-pb-model-to-keras-h5-model/

Apparently, you need to use the enclosing directory name and not the .pb filename when loading the model initially. This solves the error:

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