获取张量流模型摘要

Question

Model.summary（）给我这个输出

现在我如何检查顺序_1层和顺序_3层？ 我想要整个模型摘要，但是它给出两个顺序，所以这意味着将两个模型组合在一起，那么我如何才能获得两个模型的摘要？ 我只有model.h5个文件

Answer 1

以.h5格式保存的模型包含有关模型的所有信息。

要检查模型中模型内部的图层摘要，就像您的情况一样。 您可以提取图层，然后从每个图层中调用summary方法。

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

这是我在重现场景时使用的完整代码。

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 文件。