在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。