從 .pb 文件加載 TensorFlow 模型

Question

背景：我將我的 TensorFlow 模型存儲在包含文件夾/文件的目錄中：資產、變量和saved_model.pb。 我需要對這個模型做的就是進行預測（沒有訓練），所以我不需要資產和變量文件夾。 saved_model.pb 文件比 variables 文件夾小得多，存儲對我來說非常重要。

問題：如何在 Tensorflow 2.3 中僅從 saved_model.pb 文件（無變量或資產文件夾）加載我的 TensorFlow 模型？

這是我的文件結構：

saved_models
├── model_1
|   ├── assets
|   |   └── (this is an empty folder)
|   ├── saved_model.pb
|   ├── variables
|   |   ├── variables.data-00000-of-00001
|   |   └── variables.index

Answer 1

saved_model.pb文件只包含網絡結構的描述。 您需要變量，因為它們包含每個網絡層的權重，沒有它們，您將使用空白網絡進行預測。

Answer 2

您可以嘗試以 .h5 格式保存模型。 這只會生成一個文件。 我不確定它是否小於 .pb 大小，但試一試並找出答案。 文檔在這里。