Keras, How to get the output of each layer?

后端 未结 10 1948
借酒劲吻你
借酒劲吻你 2020-11-22 07:34

I have trained a binary classification model with CNN, and here is my code

model = Sequential()
model.add(Convolution2D(nb_filters, kernel_size[0], kernel_si         


        
10条回答
  •  野趣味
    野趣味 (楼主)
    2020-11-22 08:02

    You can easily get the outputs of any layer by using: model.layers[index].output

    For all layers use this:

    from keras import backend as K
    
    inp = model.input                                           # input placeholder
    outputs = [layer.output for layer in model.layers]          # all layer outputs
    functors = [K.function([inp, K.learning_phase()], [out]) for out in outputs]    # evaluation functions
    
    # Testing
    test = np.random.random(input_shape)[np.newaxis,...]
    layer_outs = [func([test, 1.]) for func in functors]
    print layer_outs
    

    Note: To simulate Dropout use learning_phase as 1. in layer_outs otherwise use 0.

    Edit: (based on comments)

    K.function creates theano/tensorflow tensor functions which is later used to get the output from the symbolic graph given the input.

    Now K.learning_phase() is required as an input as many Keras layers like Dropout/Batchnomalization depend on it to change behavior during training and test time.

    So if you remove the dropout layer in your code you can simply use:

    from keras import backend as K
    
    inp = model.input                                           # input placeholder
    outputs = [layer.output for layer in model.layers]          # all layer outputs
    functors = [K.function([inp], [out]) for out in outputs]    # evaluation functions
    
    # Testing
    test = np.random.random(input_shape)[np.newaxis,...]
    layer_outs = [func([test]) for func in functors]
    print layer_outs
    

    Edit 2: More optimized

    I just realized that the previous answer is not that optimized as for each function evaluation the data will be transferred CPU->GPU memory and also the tensor calculations needs to be done for the lower layers over-n-over.

    Instead this is a much better way as you don't need multiple functions but a single function giving you the list of all outputs:

    from keras import backend as K
    
    inp = model.input                                           # input placeholder
    outputs = [layer.output for layer in model.layers]          # all layer outputs
    functor = K.function([inp, K.learning_phase()], outputs )   # evaluation function
    
    # Testing
    test = np.random.random(input_shape)[np.newaxis,...]
    layer_outs = functor([test, 1.])
    print layer_outs
    

提交回复
热议问题