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