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

特尔；博士：

• 输入：Keras model变量

• 输出：一个 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：

  

  注意：如果要使用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。 但可惜只有方案，没有三维