python - 从 .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

标签： pythonfiletensorflow

Answer 1

该saved_model.pb文件仅包含网络结构的描述。您需要变量，因为它们包含每个网络层的权重，没有它们，您将使用空白网络进行预测。