如何将已加载的Keras模型的图像保存为png / jpg?

时间:2019-03-25 08:46:26

标签: python-3.x tensorflow keras

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

from keras.models import load_model

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

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

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

1 个答案:

答案 0 :(得分:2)

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

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

https://keras.io/visualization/#model-visualization