K.function用法(keras提取中间层的输出)

from keras import backend as K

1.获取名为layer_name的层的输出

#指定输出层的名称
layer_1 = K.function([model.get_input_at(0)], [model.get_layer('layer_name').output])
f1 = layer_1([image_arr])[0]

2.获取第n层的输出

#指定输出层的序号（层号从0开始）
layer_1 = K.function([model.get_input_at(0)], [model.layers[5].output])
f1 = layer_1([image_arr])[0]

3.输入输出使用layer名称

pnet = K.function([pnet.layers['data']],[pnet.layers['conv4-2'], pnet.layers['prob1']])
rnet = K.function([rnet.layers['data']],[rnet.layers['conv5-2'], rnet.layers['prob1']])
onet = K.function([onet.layers['data']],[onet.layers['conv6-2'], onet.layers['conv6-3'], onet.layers['prob1']])