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

Python 了解已保存的Tensorflow h5模型的体系结构,python,tensorflow,model,h5py,Python,Tensorflow,Model,H5py,我有一些Tensorflow模型保存为.h5文件 由于我的记录保存和文档记录不好,我想不起每一个都有什么确切的体系结构。因此,我想知道是否有一种方法,从每个模型保存的h5文件中,检查模型并确定架构 from tensorflow.keras.models import load_model model = load_model('saved_model.h5') for layer in model: try: print(layer.activation) #for so

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

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

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()
例如,有没有办法找出层数、激活函数、输入/输出、大小等

感谢您的帮助

谢谢,
Sam

根据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()

我能想到的最简单的方法是先加载模型,然后调用摘要函数no?并没有告诉我关于模型的所有信息,即激活函数,但它已经足够好了,可以使用了。谢谢你,埃德温!