如何將 .pb 文件轉換為 .h5。 （Tensorflow 模型到 keras）

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}