keras模型可视化plot_model模块调用报错

keras有模块可视化工具，可以以图片方式可视化网络结构

from keras.utils import plot_model

使用时只要指定模型和输出的文件即可

plot_model(model,to_file='model.png')

结果(model.png)如下图所示

但是

plot_model(model,to_file='model.png')

语句会报错

keras ImportError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.

该错误提示具有误导性

即便执行了

pip install pydot
pip install graphviz

仍会报错

实际上问题在于执行plot_model需要graphviz的二进制文件

解决方法是在graphviz官网下载界面http://www.graphviz.org/download/选择自己操作系统对应的版本的.msi文件

例如Windows的下载链接是https://graphviz.gitlab.io/_pages/Download/windows/graphviz-2.38.msi

安装完成后添加环境变量即可