如何从.h5和.json文件获取模型架构?

时间:2019-10-15 07:52:34

标签: keras model architecture deep-learning

我有一个经过预训练的Keras模型,带有“ .h5”和“ .json”文件。我想知道所用模型的体系结构和层的名称。反正我能做到吗?

我对此很陌生,所以我真的不知道从哪里开始

我希望加载张量流模型时会得到日志。

1 个答案:

答案 0 :(得分:0)

当然,首先加载模型,然后生成模型摘要:

from keras.models import load_model

model = load_model('your_model.hdf5')

model.summary()

摘要将包含图层名称和输入/输出形状。