Windows7下Keras使用plot_model进行模型可视化

一、基本环境

Windows7、Anaconda3

二、安装pydot和GraphViz

1.安装GraphViz

       在cmd中输入conda install graphviz,安装GraphViz。

2.安装GraphViz软件本身

       进入https://graphviz.gitlab.io/_pages/Download/Download_windows.html下载graphviz-2.38.msi,解压后安装。

Windows7下Keras使用plot_model进行模型可视化_第1张图片

       进入GraphViz安装路径,找到bin文件路径,我的是D:/Graphviz/bin/。将该路径添加到环境变量(我是用户变量和系统变量都添加了)。

        如果后面报了错误,可能是手动添加环境变量不起作用,就在Spyder中运行下列程序:

import os

os.environ["PATH"] += os.pathsep + 'D:/Graphviz/bin/'

3.安装pydot

       在cmd中运行conda install pydot,安装pydot。

4.安装顺序不能变,按步骤安装即可。

三、使用plot_model

from keras.utils import plot_model


# show_shapes:指定是否显示输出数据的形状,默认为False;show_layer_names:指定是否显示层名称,默认为True
modelstruct_path='C:/Users/Administrator/.keras/model.png'  # 路径
plot_model(model, to_file=modelstruct_path,show_shapes=True,show_layer_names=True)   # 保存模型

 

你可能感兴趣的:(深度学习)