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对象生成这样的图表文件是完全可能的。

特尔；博士：

• 输入：一个 Kerasmodel变量

• 输出：一个 PNG 图像

---

PS：

• 我已经知道这个在线工具：http : //alexlenail.me/NN-SVG/LeNet.html和这个问题如何绘制深度学习网络架构图？但这里的想法是从 Keras 自动生成它。

• 相关但不相同：您如何可视化神经网络架构？ .

• 不同于如何使用 Keras 可视化神经网络架构？

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

• 链接文章：卷积神经网络的简单图

• 做from keras.utils import plot_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。但遗憾的是只有方案，没有三维