如何加載 keras model 另存為.pb

Question

我想加載一個我訓練過的 keras model 並將其保存為.pb 。
這是代碼，

我正在使用一個 jupyter 筆記本。
model成功保存為同目錄下的saved_model.pb 。 但是代碼無法訪問它。

任何人都可以看到它，我如何訪問保存在.pb擴展名中的這個 keras model。
我在其他幾個地方檢查了解決方案，但沒有運氣。

Model 保存在model/saved_model.pb中。
我已取出.pb文件並將其放在我的代碼文件所在的同一目錄中。

Answer 1

function tf.keras將tf.keras.models.load_model加載到SavedModel -model 中。 function 的參數是保存的 model 的路徑。

所以試試model = tf.keras.models.load_model('model')

Answer 2

您應該加載所有 model 文件夾而不是 loading.pb 文件。 如果將 model 保存到'./_models/vgg50_finetune' （我在項目中使用了此路徑），您將獲得包含兩個.pb 文件（keras_metadata.pb 和 saved_model.pb）和兩個子文件夾（資產和變量）的文件夾 vgg50_finetune。 你唯一應該做的就是使用這個代碼： model = tf.keras.models.load_model('./_models/vgg50_finetune')你可以訓練 Z20F35E630DAF44DFACZF 或使用它來預測8F44DB84ZC3