как сохранить изображение загруженной модели keras в формате png / jpg?

#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 или требуется использование другой библиотеки?

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

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