Keras获取中间某一层的输出
搭建模型骨架,即指定输入输出;
load模型的训练参数;
构建中间模型,指定输出层;
whole_model = Model(inputs = model_input,outputs = model_output)
whole_model.load_weights(model_saved_path) #模型保存格式为.h5格式,或其他,要与load方式一一对应
middle_model = Model(inputs = model_input,outputs = whole_model.get_layer('channel_layer').output)
middle_output = middle_model.predict(test_image,batch_size=batch_size)
最重要的是whole_model.get_layer('channel_layer').output
。