keras模型可视化环境配置

首先是要安装依赖项

pip install pydot_ng
pip install graphviz

然后是安装graphviz

到 graphviz网站 下载graphviz-2.38.msi文件，并安装。将安装路径C:...\graphviz\bin加到环境变量里。（这里一定要加，不然会报错GraphViz's executables not found : Anaconda-3）

修改Keras源码

这个时候还是会报错：

ImportError: Failed to import `pydot`. Please install `pydot`. For example with `pip install pydot`.

这个是Keras的问题，要将路径C:\Anaconda3\lib\site-packages\keras\utils下的vis_utils.py文件中的代码

try:
    import pydot
except ImportError:
    pydot = None

替换为：

try:
    import pydot_ng as pydot
except ImportError:
    pydot = None

OK啦~

 

 

然后可视化模型很简单

from keras.utils import plot_model

#**************前面是你定义的模型************
model = yourmodel()
plot_model(model,to_file='D:\\model.png',show_shapes=True)#就是这一句话