Создать диаграмму архитектуры модели CNN в Керасе

Question

Я следовал руководству, чтобы создать визуально графическое представление модели CNN, используя это: https://keras.io/visualization/

Мой код на данный момент выглядит следующим образом:

from keras.utils import plot_model
from keras.applications.resnet50 import ResNet50
import numpy as np

model = ResNet50(weights='imagenet')
plot_model(model, to_file='model.png')

Когда яС помощью вышеупомянутого кода я могу создать графическое представление (используя Graphviz) ResNet50 и сохранить его в «model.png».Но я хочу создать структурную схему модели CNN со слоями.Вот пример моего желаемого результата:

Любая идея, как я могу достичь вышеупомянутой блок-схемы программно, а не просто генерировать диаграмму диаграммы модели CNN?

Answer 1

Вот полный список существующих решений (не только для Keras): Как вы визуализируете архитектуры нейронных сетей? .

Обратите внимание на Netron .