windows和linux环境下keras的模型框架可视化

1.简介

keras提供了模型可视化模块,下面讲解下安装教程和简易教程。

2.安装教程

2.1windows环境下的安装

2.1.1安装指定模块

pip install pydot-ng
pip install graphviz
pip install pydot==1.2.3 

2.1.2安装辅助应用程序

安装graphviz-2.38.msi,直接下一步即可,并且将安装路径C:\Program Files (x86)\Graphviz2.38\bin添加到path中

文件下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html

注:2019/2/25的pydot最新版本为1.4.1,直接安装最新版会报错,需要注意。

2.2linux环境下的安装

2.2.1安装指定模块

pip install pydot-ng 
pip install graphviz
pip install pydot

2.2.2安装辅助应用程序

使用apt install graphviz,yum install graphviz等命令安装graphviz程序,具体命令依据自身的linux系统

3.示例教程

 from keras.models import Model
from keras.utils.vis_utils import plot_model
from keras.layers import Input, Lambda, concatenate # 切片函数,这里在最后一维切
def slice(x,index):
return x[:,:,:,index] # a相当于规模未知的三维格式的数据
a = Input(shape=(5,6,2))
a1 = Lambda(slice, output_shape=(5,6), arguments={'index':0})(a)
a2 = Lambda(slice, output_shape=(5,6), arguments={'index':1})(a)
output = concatenate([a1, a2], axis=2)
model = Model(a, output)
plot_model(model, to_file='./my_visual.png', show_shapes=True)
print(a.shape)
print(a1.shape)
print(a2.shape)

图示如下:

windows和linux环境下keras的模型框架可视化

!!!

上一篇:nodejs异常处理过程/获取nodejs异常类型/写一个eggjs异常处理中间件


下一篇:开始学nodejs —— 调试篇