Функция Keras или Tensorflow для построения трехмерной диаграммы структуры нейронной сети?

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 (или другой библиотеке) функция для автоматического создания трехмерной диаграммы структуры? , например:

или

или

Генерация такого файла диаграммы была бы полностью возможной из объекта model.

TL; DR:

• ВХОД: a Keras model переменная

• ВЫХОД: изображение PNG

---

PS:

• Я уже знаете этот онлайн-инструмент: http://alexlenail.me/NN-SVG/LeNet.html и этот вопрос Как рисовать диаграммы архитектуры сети с глубоким обучением? но здесь идея состоит в том, чтобы сгенерировать это автоматически из Keras.

• Связаны, но не идентичны: Как вы визуализируете архитектуры нейронных сетей? .

• Отличается от Как можно представить архитектуру нейронной сети с помощью Keras?

• Связанный проект: https://github.com/stared/keras-sequential-ascii

• Связанный артикул: Простые диаграммы извилистых нейронных сетей

• Doing 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. Но, к сожалению, только схема, а не трехмерная