在 Keras 中创建 CNN 模型架构图答案

【问题标题】：Create CNN model architecture diagram in Keras在 Keras 中创建 CNN 模型架构图
【发布时间】：2019-03-01 11:02:36
【问题描述】：

我按照教程使用以下方法创建 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 模型的图表吗？

【问题讨论】：

看看这个很酷的回购：github.com/HarisIqbal88/PlotNeuralNet
我试过了。不幸的是，它目前不起作用，并且还提出了与此相关的问题。

标签： tensorflow keras deep-learning conv-neural-network graphviz

【解决方案1】：

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

关注Netron。

【讨论】：

感谢您的回复。在我使用 Netron 之前问一个简单的问题：我可以将最终的框图下载为 png 或 pdf 格式吗？我已经尝试过 Netscope 之一，但我似乎无法将网络摘要/图表下载为图像文件。
很遗憾，我没有使用它。但你总是可以截图:)