Tensorflow2如何获取中间层输出

Question

我想用tensorflow2实现cnn提取图像特征然后输出到SVM进行分类。有哪些好的方法呢？我查了tensorflow2的文档，没有很好的解释。谁能指导我？

Answer 1

感谢您上面的澄清答案。我之前写过类似问题的回答。但是您可以通过使用所谓的功能模型 API构建辅助模型来从 tf.keras 模型中提取中间层输出。“函数模型 API”使用tf.keras.Model()。当您调用该函数时，您指定参数inputs和outputs。您可以在参数中包含中间层的输出outputs。请参阅下面的简单代码示例：

import tensorflow as tf

# This is the original model.
model = tf.keras.Sequential([
    tf.keras.layers.Flatten(input_shape=[28, 28, 1]),
    tf.keras.layers.Dense(100, activation="relu"),
    tf.keras.layers.Dense(10, activation="softmax")])

# Make an auxiliary model that exposes the output from the intermediate layer
# of interest, which is the first Dense layer in this case.
aux_model = tf.keras.Model(inputs=model.inputs,
                           outputs=model.outputs + [model.layers[1].output])

# Access both the final and intermediate output of the original model
# by calling `aux_model.predict()`.
final_output, intermediate_layer_output = aux_model.predict(some_input)