From f92546b56e456846b1ea680722495f5f0f977a61 Mon Sep 17 00:00:00 2001 From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de> Date: Fri, 31 Aug 2018 12:11:14 +0200 Subject: [PATCH] revert read_activations.py back to process all layers --- keras_visualize_activations/read_activations.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/keras_visualize_activations/read_activations.py b/keras_visualize_activations/read_activations.py index b5d34ab..0e4641a 100644 --- a/keras_visualize_activations/read_activations.py +++ b/keras_visualize_activations/read_activations.py @@ -1,6 +1,5 @@ import keras.backend as K -from keras.layers import Masking, InputLayer def get_activations(model, model_inputs, print_shape_only=False, layer_name=None): print('----- activations -----') @@ -13,12 +12,8 @@ def get_activations(model, model_inputs, print_shape_only=False, layer_name=None inp = [inp] model_multi_inputs_cond = False - # all layer outputs - # skip input and masking layers outputs = [layer.output for layer in model.layers if - (layer.name == layer_name or layer_name is None) - and not isinstance(layer, InputLayer) - and not isinstance(layer, Masking)] + layer.name == layer_name or layer_name is None] # all layer outputs funcs = [K.function(inp + [K.learning_phase()], [out]) for out in outputs] # evaluation functions -- GitLab