TF2.0#
https://tensorflow.google.cn/tutorials/images/classification
一个简单的方法是创建一个新的 Model 来输出你所感兴趣的层:
from keras.models import Model
model = ... # 创建原始模型
layer_name = 'my_layer'
intermediate_layer_model = Model(inputs=model.input,
outputs=model.get_layer(layer_name).output)
intermediate_output = intermediate_layer_model.predict(data)
或者,你也可以构建一个 Keras 函数,该函数将在给定输入的情况下返回某个层的输出,例如:
from keras import backend as K
get_3rd_layer_output = K.function([model.layers[0].input],
[model.layers[3].output])
layer_output = get_3rd_layer_output([x])[0]