了解已保存的Tensorflow h5模型的体系结构

Question

我有一些Tensorflow模型保存为.h5文件

由于我的记录保存和文档记录不好，我想不起每一个都有什么确切的体系结构。因此，我想知道是否有一种方法，从每个模型保存的h5文件中，检查模型并确定架构

例如，有没有办法找出层数、激活函数、输入/输出、大小等

感谢您的帮助

谢谢， 萨姆

Answer 1

根据Edwin在评论中的建议，您可以加载模型并查看层详细信息的摘要。
您可以使用下面的代码来获取有关激活功能和模型体系结构的信息

from tensorflow.keras.models import load_model

model = load_model('saved_model.h5')

for layer in model:
  try:
    print(layer.activation)

  #for some layers there will not be any activation fucntion.
  except:
    pass

#To get the name of layers in the model.
layer_names=[layer.name for layer in model.layers]

#for model's summary and details.
model.summary()