首页 > 其他 > 详细

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

时间:2019-08-29 14:43:24      阅读:515      评论:0      收藏:0      [点我收藏+]

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)

结果如图所示:

 

技术分享图片

 

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

原文:https://www.cnblogs.com/travelcat/p/11429437.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!