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

Answer 1

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

关注内创。