keras绘制网络结构

工具

• pydot_ng
  注意:如果是安装pydot而不是pydot_ng，哪怕安装成功graphviz还是会报错:

IMPORTERROR : FAILED TO IMPORT PYDOT. YOU MUST INSTALL PYDOT AND GRAPHVIZ FOR PYDOTPRINT TO WORK.

# 直接使用pip安装
pip install pydot_ng

• graphviz

# 不能直接pip，从下面网站下载，下载之后一直点击next，之后添加到环境变量中去
https://www2.graphviz.org/Packages/stable/windows/10/cmake/Release/x64/

下载

keras绘制网络结构

from keras.utils.vis_utils import plot_model
import matplotlib.pyplot as plt
# 保存结构
plot_model(model, to_file='model.png', show_shapes=True, show_layer_names=False, rankdir='TB') # TB从上往下，LR从左往右
# 读取图片
plt.figure(figsize=(10, 10))
img = plt.imread('model.png')
plt.imshow(img)
plt.axis('off')
plt.show()

网络结构显示效果: