一、报错提示

调用keras.utils.plot_model报错提示:
('You must install pydot (pip install pydot) and install graphviz (see instructions at https://graphviz.gitlab.io/download/) ', ‘for plot_model/model_to_dot to work.’)

因为tensoerflow2.0以上兼容了keras,只是产生图而已,所以不必在这里使用tf.keras,再者使用tf.keras也会报错,tf的库里没有这个函数。

二、解决方案(共四步)

2.1 安装pydot

按照提示,安装pydot

pip install pydot

2.2 下载并安装graphviz包

踩坑:千万别pip install graphviz。因为测试过无效。
正确做法是从官网
找到如下图所示的Win10 64位版本的graphviz进行下载,然后安装
在这里插入图片描述
在这里插入图片描述
下一步就好了
在这里插入图片描述

2.3 配置环境变量(graphviz)

进入到安装的bin目录下
在这里插入图片描述
在这里插入图片描述
变量名:别与系统的专用词冲突即可
变量值:路径,我这里是D:\Graphviz\bin (仅供参考)

2.4 调用

添加环境变量后,引入包

from keras.utils.vis_utils import plot_model

模型编译后调用plot_model函数(见下图)
(to_file中设置产生图片的路径,下面这种方式表示保存到当前文件下)

keras.utils.plot_model(model, to_file='graph.png', show_shapes=True)

在这里插入图片描述

三、结果展示

模型编译后运行就会产生对应的可视化结构

在这里插入图片描述
由于比较长,只截取部分

在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐