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)

图示如下：

！！！