使用keras绘制网络结构图时出现'`pydot` failed to call GraphViz.'错误的解决办法

初次尝试绘制网络结构图

笔者使用的是Anaconda,因此在Anaconda Prompt中安装了pydot以及graphviz(pip install pydot & pip install graphviz)后 ,打开Python绘制网络图,结果程序报错了。
在这里插入图片描述

第二次尝试

找到了一篇博客,发现原来还需要手动下载Graphviz,不只是pip一下就可以的。于是参考了下面这篇博文,下载安装Graphviz后,把安装文件夹中的bin文件夹对应的路径添加到path环境变量中。
参考: 手动下载Graphviz.
参考: win10添加环境变量.
Graphviz下载安装了,环境变量也添加了,重新运行程序依然报错,错误依然是pydot无法找到GraphViz。
笔者再次查找原因,原来问题出在这里。

第三次尝试

在国外一个论坛上找到了答案,原来是1.2.x版本的pydot已经弃用find_graphviz()了,因此笔者转而使用pydot_ng。
在这里插入图片描述
在程序中import pydot_ng as pydot后,本以为应该没什么问题了,结果依然报错。。。这时在网上找遍各个地方依然没有答案,自己想了想才恍然大悟。

最后的尝试

原因在于,虽然笔者程序中import的是pydot_ng,但keras源码导入的依然是pydot,因此我们需要稍微修改一下keras中vis_utils.py的代码。
在Anaconda的安装目录下,即Anaconda3\lib\site-packages\keras\utils\vis_utils.py中找到"import pydot",把它改成"import pydot_ng as pydot",plot_model一下网络结构图便正确绘制完成了。
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/u013899126/article/details/88319505