python - Keras 或 Tensorflow 函数绘制神经网络结构的 3D 图？

Question

使用 Keras 创建神经网络时:

model = Sequential()
model.add(Conv2D(32, kernel_size=(3, 3), activation='relu', input_shape=input_shape))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
...

函数model.summary有助于了解结构的整体情况。

为了更好地了解结构，Keras 或 Tensorflow(或其他库)中是否有函数可以自动生成结构的 3D 图？如下所示:

或

或

完全可以从 model 对象生成这样的图表文件。

TL；博士:

• 输入:Keras 模型变量

• 输出:PNG 图像

<小时/>

附注:

• 我已经知道这个在线工具:http://alexlenail.me/NN-SVG/LeNet.html还有这个问题How to draw Deep learning network architecture diagrams?但这里的想法是从 Keras 自动生成它。

• 链接但不相同:How do you visualize neural network architectures? .

• 与 How can a neural network architecture be visualized with Keras? 不同

• 链接项目:https://github.com/stared/keras-sequential-ascii

• 链接文章:Simple diagrams of convoluted neural networks

• 执行 from keras.utils importplot_model plot_model(model, to_file='model.png', show_shapes=True, show_layer_names=True) 已经给出了一些东西但它不是 3D:

  

  注意:对于 Windows 添加此:os.environ["PATH"] += os.pathsep + r'C:\Program Files (x86)\Graphviz2.38\bin' 安装后Graphviz 如果您想使用plot_model。

Answer 1

你在谈论这个吗: https://keras.io/visualization/

from keras.utils import plot_model
plot_model(model, to_file='model.png')

这会将输入和输出张量的模型结构保存为 png。但遗憾的是只有方案，没有三维