Tensorflow+keras用代码给神经网络结构绘图

Posted 2021-06-01 Better Bench

1 引言

在论文中可能有必要讲述清楚网络的结构，可以用代码实现漂亮的绘制结构。如图所示。

2 步骤

安装两个包，本人是LInux系统，用以下命令，如果是其他系统，参考网址

sudo apt install graphviz
pip install pydot

3 实现

import tensorflow as tf
from tensorflow.keras.layers import *
from tensorflow.keras.models import Sequential, Model

model = tf.keras.Sequential()
model.add(Conv1D(8,5,strides=2,padding='same',input_shape=(4,4,8)))
model.add(LeakyReLU())
model.add(Dropout(0.3))

model.add(Conv1D(16,5, strides=2, padding='same'))
model.add(LeakyReLU())
model.add(Dropout(0.3))
model.add(Flatten())
model.add(Dense(1,activation='softmax'))
tf.keras.utils.plot_model(model, to_file='model.png', show_shapes=True)