python-3.x - 如何将加载的 keras 模型的图像保存为 png/jpg？

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

from keras.models import load_model

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

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

有没有办法在 keras 中执行此操作，或者是否需要使用另一个库？

标签： python-3.xtensorflowkeras

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

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