在Keras中创建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')

当我使用上述代码时，我能够创建 ResNet50 的图形表示（使用 Graphviz）并将其保存在 'model.png' 中。但我想用图层创建 CNN 模型的框图。我想要的输出示例如下：

知道如何以编程方式实现上述框图，而不仅仅是生成 CNN 模型的图表吗？

Answer 1

以下是现有解决方案的完整列表（不仅适用于 Keras）：How do you visualize neural network architectures?.

关注Netron.

在Keras中创建CNN模型架构图

Create CNN model architecture diagram in Keras

graphviz

deep-learning

conv-neural-network

keras

tensorflow