如何加载保存为 .pb 的 keras 模型

Question

我想加载一个我训练过的 keras 模型并将其保存为.pb。
这是代码，

我正在使用 jupyter 笔记本。
模型成功保存为同一目录下的saved_model.pb。但是代码无法访问它。

任何人都可以看到它，我如何访问保存在.pb扩展名中的这个 keras 模型。
我在其他几个地方检查了解决方案，但没有运气。

模型保存在model/saved_model.pb。
我已经取出.pb文件并将其放在我的代码文件所在的同一目录中。

Answer 1

该函数将 atf.keras.models.load_model加载SavedModel到tf.keras-model 中。该函数的参数是保存模型的路径。

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

Answer 2

您应该加载所有模型文件夹而不是加载 .pb 文件。如果将模型保存到'./_models/vgg50_finetune'（我在项目中使用了此路径），您将获得文件夹 vgg50_finetune，其中包含两个 .pb 文件（keras_metadata.pb 和 saving_model.pb）和两个子文件夹（资产和变量）。您唯一应该做的就是使用此代码： model = tf.keras.models.load_model('./_models/vgg50_finetune') 您可以训练模型或使用它进行预测。