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)

结果如图所示:

猜你喜欢

转载自www.cnblogs.com/travelcat/p/11429437.html