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的模型框架可视化 - Python技术站