我按照该教程使用以下方法创建了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模型的图形图?
答案 0 :(得分:0)
以下是现有解决方案的完整列表(不仅适用于Keras):How do you visualize neural network architectures?。
请注意Netron。