想要导出模型中间层的输出,发现实际上就是建立一个子模型,将某一特定层进行输出即可
参考:
intermediate_layer = tf.keras.models.Model(
inputs=model.input,
outputs=model.get_layer(name='tf.concat').output
)
想要导出模型中间层的输出,发现实际上就是建立一个子模型,将某一特定层进行输出即可
参考:
intermediate_layer = tf.keras.models.Model(
inputs=model.input,
outputs=model.get_layer(name='tf.concat').output
)