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,解压后安装。

       进入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)   # 保存模型

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值