我使用 MaskRCNN 训练了我自己的数据集。我训练的文件是 .h5 文件格式。现在我想知道/读取输入节点名称和输出节点名称。
我该如何阅读？
如何从 keras 模型中的 .h5 文件中读取输入节点和输出节点？
数据挖掘
Python
深度学习
喀拉斯
张量流
物体检测
2022-03-14 20:39:51
1个回答
我假设您已准备好 Keras .h5 文件。现在，我们需要使用从该文件加载模型，
model = keras.models.load_model( 'model.h5' )
model代表 Keras 模型。现在，model.input将对应于输入张量。
input = model.input
output = model.output
此外，您以类似的方式获得输出。input和张量将output有它们的名字。这可以通过使用 Python 打印张量来观察print。
其它你可能感兴趣的问题