Créer un diagramme d'architecture de modèle CNN en Keras

Question

J'ai suivi le tutoriel pour créer une représentation graphique visuelle du modèle CNN en utilisant ceci : https://keras.io/visualization/

Mon code pour le moment est le suivant :

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')

Lorsque j'utilise le code mentionné ci-dessus, je suis en mesure de créer une représentation graphique (en utilisant Graphviz) de ResNet50 et de l'enregistrer dans 'model.png'. Mais je veux créer un diagramme en bloc du modèle CNN avec les couches à la place. Un exemple de la sortie souhaitée est le suivant :

Avez-vous une idée de comment je peux réaliser le diagramme en bloc mentionné de manière programmatique plutôt que de simplement générer le diagramme graphique du modèle CNN ?

Answer 1

Voici une liste exhaustive des solutions existantes (pas seulement pour Keras) : Comment visualisez-vous les architectures de réseaux neuronaux ?.

Prêtez attention à Netron.