keras模型框架以图片的形式显示

原文网址：http://blog.csdn.net/u012313895/article/details/79075750

首先安装keras2的三个可视化库

sudo pip/pip3 install pydot
sudo pip/pip3 install pydot_ng
sudo apt-get install graphviz

然后运行下面代码生成模型框架图

from model import VGG16  # VGG16已经在model.py文件中定义好了
from keras.utils.vis_utils import plot_model
model1 = VGG16(include_top=False, weights=None,input_shape=(320, 320, 3), pooling=None)  # 这里将函数式模型实例化
plot_model(model1, to_file='model_vgg16.jpg', show_shapes=True)

运行后生成的模型框架图如下所示：