Skip to content
Snippets Groups Projects
Commit 45f0ff96 authored by Nikolai's avatar Nikolai
Browse files

skip InputLayer and MaskingLayer in read_activations

parent c4c88456
No related branches found
No related tags found
No related merge requests found
import keras.backend as K import keras.backend as K
from keras.engine.input_layer import InputLayer
from keras.layers.core import Masking
def get_activations(model, model_inputs, print_shape_only=False, layer_name=None): def get_activations(model, model_inputs, print_shape_only=False, layer_name=None):
print('----- activations -----') print('----- activations -----')
...@@ -12,8 +14,12 @@ def get_activations(model, model_inputs, print_shape_only=False, layer_name=None ...@@ -12,8 +14,12 @@ def get_activations(model, model_inputs, print_shape_only=False, layer_name=None
inp = [inp] inp = [inp]
model_multi_inputs_cond = False model_multi_inputs_cond = False
# all layer outputs
# skip input and masking layers
outputs = [layer.output for layer in model.layers if outputs = [layer.output for layer in model.layers if
layer.name == layer_name or layer_name is None] # all layer outputs (layer.name == layer_name or layer_name is None)
and not isinstance(layer, InputLayer)
and not isinstance(layer, Masking)]
funcs = [K.function(inp + [K.learning_phase()], [out]) for out in outputs] # evaluation functions funcs = [K.function(inp + [K.learning_phase()], [out]) for out in outputs] # evaluation functions
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment