如何将加载的 keras 模型的图像保存为 png/jpg？答案

【问题标题】：how to save image of loaded keras model as png/jpg?如何将加载的 keras 模型的图像保存为 png/jpg？
【发布时间】：2019-08-15 11:19:00
【问题描述】：

我已经训练了一个 keras 模型并将其保存以供以后进行预测。但是，我使用以下方法加载了保存的模型：

from keras.models import load_model

#Restore saved keras model
restored_keras_model = load_model("C:/*******/saved_model.hdf5")

现在我想保存加载模型的图像，以便在使用它之前进行可视化，然后再进行预测。

有没有办法在 keras 中做到这一点，还是需要使用另一个库？

【问题讨论】：

【解决方案1】：

是的，除了执行 restore_keras_model.summary() 之外，您还可以使用 plot_model API 将模型架构保存为 png 文件。

from keras.utils import plot_model
plot_model(restored_keras_model, to_file='model.png')

【讨论】：