TensorFlow/Keras保存模型的网络架构图片

• 第一步：在python中安装pydot库

pip install pydot

• 第二步：在机器中安装GraphViz

下载网址：https://graphviz.org/download/

安装完成后，在环境变量的path中添加graphviz的安装路径下bin目录，D:\graphviz\bin\

• 第三步：添加python程序

from keras.utils import plot_model
plot_model(model, to_file='model.png',show_shapes=True)

• 最后，输出如下：