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.示例教程

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

图示如下:

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

!!!