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)
图示如下:
!!!