tensorflow - 在 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 模型的图表吗？

标签： tensorflowkerasdeep-learningconv-neural-networkgraphviz

Answer 1

以下是现有解决方案的完整列表（不仅适用于 Keras）：您如何可视化神经网络架构？.

关注Netron。