TensorFlow2可视化网络层数（keras中的plot_model模块）

今天在看论文的时候发现有keras中的plot_model模块进行可视化，效果还不错

安装说明

先装Graphviz，再装pydot_ng

Graphviz

1、到官网下载graphviz的安装包 https://graphviz.org/download/


记得勾选这个选项

2、测试：
在cmd中输入 dot -version并回车，若显示出graphviz的相关版本信息，则安装配置成功。

dot -version

3、在终端中使用命令行安装graphviz：

pip install graphviz

pydot_ng

1、安装pydot_ng。使用pip安装：

pip install pydot_ng

2、测试（测试之前先重启哦）

import pydot_ng as pydot
print (pydot.find_graphviz())

显示结果

将问号改为None

Windows找这个目录文件
Anaconda\envs\yourEnv\Lib\site-packages\tensorflow\python\keras\utils\vis_utils.py

问题出在vis_utils.py文件中下面这个函数上面

def format_shape(shape):
  return str(shape).replace(str(None), '?')

直接改为：

def format_shape(shape):
  return str(shape).replace(str(None), 'None')

使用plot_model模块

tf.keras.utils.plot_model(model, show_shapes=True,to_file='picture/LSTM模型/model.png')

show_shapes=True必须加
效果图：还不错