Keras:找出层数

Posted

技术标签:

【中文标题】Keras:找出层数【英文标题】:Keras: find out the number of layers 【发布时间】:2018-06-24 06:10:30 【问题描述】:

有没有办法获取 Keras 模型中的层数(不是参数)?

model.summary() 信息量很大,但从中获取层数并不简单。

【问题讨论】:

【参考方案1】:

model.layers 会给你所有层的列表。因此这个数字是len(model.layers)

【讨论】:

len(model.layers) 无法计算“子层”,即模型中存在的层,充当您的 ubermodel 中的层。【参考方案2】:

要获得图层的图形视图,您可以使用: from keras.utils.vis_utils import plot_model plot_model(model, to_file='layers_plot.png', show_shapes=True, show_layer_names=True)

您需要pip install pydot 并从https://graphviz.gitlab.io/download/ 下载并安装graphviz 附上一个 sample output image

【讨论】:

以上是关于Keras:找出层数的主要内容,如果未能解决你的问题,请参考以下文章

Keras网络层之“关于Keras的层(Layer)”

Keras:密集层和激活层之间的形状不匹配

Keras 序列模型输入层

Keras网络层之常用层Core

keras模块学习之层(layer)的使用-笔记

Keras网络层之卷积层