首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >tensorflow模型综述

tensorflow模型综述
EN

Stack Overflow用户
提问于 2020-02-20 16:46:34
回答 1查看 2.8K关注 0票数 1

Model.summary()给我一个输出

现在如何检查sequential_1层和sequential_3层?我想要整个模型摘要,但它给出了两个顺序,这意味着两个模型是结合在一起的,那么如何得到这两个模型的总结呢?我只有model.h5文件

EN

回答 1

Stack Overflow用户

发布于 2020-03-25 07:49:36

格式保存的.h5模型包括有关该模型的所有内容.

检查模型中模型中的图层摘要,就像在您的例子中一样。您可以提取这些层,然后从每个层调用汇总方法。

即。layer_summary = [layer.summary() for layer in loaded_model.layers]

下面是我在复制场景时使用的完整代码。

代码语言:javascript
运行
复制
import tensorflow as tf
print('Running Tensorflow version {}'.format(tf.__version__)) # Tensorflow 2.1.0

model_path = '/content/keras_model.h5'

loaded_model = tf.keras.models.load_model(model_path)
loaded_model.summary()

inp = loaded_model.input   
layer_summary = [layer.summary() for layer in loaded_model.layers]   

我还使用了您上传的model.h5文件。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/60324855

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档