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

Question

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

from keras.models import load_model

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

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

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

Answer 1

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

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