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

最新推荐文章于 2024-05-25 09:54:36 发布

转载最新推荐文章于 2024-05-25 09:54:36 发布 · 1.6k 阅读

文章标签：

#keras模型框架图 #模型图 #框架图 #keras模型

keras 专栏收录该内容

5 篇文章

订阅专栏

本文介绍如何使用Keras2及配套可视化库生成模型框架图，包括安装所需库、定义及实例化模型、调用绘图函数等步骤。

原文网址：http://blog.youkuaiyun.com/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)