我按照教程创建了 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 模型的图表吗?
最佳答案
以下是现有解决方案的完整列表(不仅适用于 Keras):How do you visualize neural network architectures? .
关注Netron .
关于tensorflow - 在Keras中创建CNN模型架构图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54943307/