Keras.utils.vis_utils.plot_model 绘制神经网络结构示意图

keras的内置函数keras.utils.vis_utils.plot_model可以直接将搭建的神经网络用流程图表示出来

 

该函数原来为keras.utils.visualize_util.plot，在2017年3月1日的更新中作了修改

visualize_util改为vis_utils；plot改为plot_model；

使用原函数名会报错import error

 

使用该函数需要额外的函数库和软件，以下说明引自博客：

https://www.jianshu.com/p/3a97c0a43beb

1. （使用pip install）安装pydot和graphviz两个函数库

2. 安装graphviz软件：

https://graphviz.gitlab.io/_pages/Download/Download_windows.html

3. 将路径加入系统变量，从我的电脑-属性-高级系统设置-环境变量 进入

（图片为引用，根据安装路径调整）

4. 检查graphviz是否安装成功，进入windows命令行，输入命令：dot -version，若成功显示版本信息，则认为安装和添加路径成功

5. 引用函数需要以下语句

from keras.utils.vis_utils import plot_model  

import os

os.environ["PATH"] += os.pathsep +'D:/Program Files (x86)/Graphviz2.38/bin/' # 后一段路径改为自己安装的路径

6. 绘图语句如下：

plot_model(model,to_file='name.png',show_shapes=True)

结果如图所示：

 

 

转载于:https://www.cnblogs.com/travelcat/p/11429437.html