在Keras中创建CNN模型架构图

时间:2019-03-01 11:02:37

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

我按照该教程使用以下方法创建了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模型的框图。我期望的输出示例如下: enter image description here

有什么想法能以编程方式实现上述框图,而不仅仅是生成CNN模型的图形图?

1 个答案:

答案 0 :(得分:0)

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

请注意Netron