从.pb文件加载TensorFlow模型

Question

背景：我将TensorFlow模型存储在一个目录中，其中包含以下文件夹/文件：资产，变量和saved_model.pb。我需要做的就是对该模型进行预测（无需培训），因此不需要资产和变量文件夹。 saved_model.pb文件比变量文件夹小得多，存储对我来说非常重要。

问题：如何仅从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大小，但请尝试一下。文档为here.