Как сохранить изображение загруженной модели Keras в формате PNG / JPG?

Question

Я обучил модель keras и сохранил ее, чтобы потом делать прогнозы.Однако я загрузил сохраненную модель, используя:

from keras.models import load_model

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

Теперь я хотел бы сохранить изображение загруженной модели, чтобы я мог визуализировать его перед использованием, прежде чем делать прогнозы.

Есть лиспособ сделать это в керасе или требуется использование другой библиотеки?

Answer 1

Да, в добавление к restore_keras_model.summary (), вы можете сохранить архитектуру модели в виде png-файла, используя API plot_model.

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

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