Skip to content
Snippets Groups Projects
Commit f92546b5 authored by Nikolai.Hartmann's avatar Nikolai.Hartmann
Browse files

revert read_activations.py back to process all layers

parent d94eca34
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.layers import Masking, InputLayer
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 -----')
...@@ -13,12 +12,8 @@ def get_activations(model, model_inputs, print_shape_only=False, layer_name=None ...@@ -13,12 +12,8 @@ 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) layer.name == layer_name or layer_name is None] # all layer outputs
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