Keras 查看模型结构并保存图片
安装
首先安装pydot 和 graphviz ,尽量采用conda进行安装,可以比较方便的解决依赖的问题
conda install pydot, graphviz
使用
此方法相对于网络的直接保存png,图像更清晰,可以在VS Code 安装插件Svg Preview(Simon Siefke),方便预览
from keras.utils.vis_utils import model_to_dot
src = model_to_dot(model,show_shapes=True,
show_layer_names=True,
rankdir='TB',
dpi=300, expand_nested=True,subgraph = False)
src.write_svg("test.svg")
对于Keras 的可视化可以再参考:Keras可视化