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;DR：

• 输入：一个 Keras model 变量

• 输出：PNG 图像

---

PS：

• 我已经知道这个在线工具：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 import plot_modelplot_model(model, to_file='model.png', show_shapes=True, show_layer_names=True)已经给出了一些东西，但它不是3D：

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

Answer 1

你是在说这个吗： https://keras.io/visualization/

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

这会将带有输入和输出张量的模型结构保存为 png。 但可惜只有方案，不是立体的