在Keras 中文文档中可以看到,使用keras.utils.vis_utils模块绘制Keras模型,
具体实现代码如下:
from keras.utils import plot_model
plot_model(model, to_file='model.png') #导出model模型图到程序所在路径下的model.png中
plot_model 有 4 个可选参数:
show_shapes (默认为 False) 控制是否在图中输出各层的尺寸。 show_layer_names (默认为 True) 控制是否在图中显示每一层的名字。 expand_dim(默认为 False)控制是否将嵌套模型扩展为图形中的聚类。 dpi(默认为 96)控制图像 dpi。
调用pydot的报错解决方法
环境: windows10、Python3.6、Keras2.3.0
1. 安装pydot
由于Keras中使用了pydot.find_graphviz()函数,而这个函数在1.2.*之后的版本中被剥离了,所以不能直接安装pydot,而是安装pydot_ng。
1.1 安装pydot_ng
pip install pydot_ng
1.2 报错处理
在运行时出现NameError: name ‘pydot’ is not defined的错误,点击"C:\Python安装路径\Python\Python36\lib\site-packages\keras\utils\vis_utils.py"
或者直接在Python安装路径打开vis_utils.py文件