从 .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 大小，但试一试并找出答案。 文档在这里。