Fixes #45 (KerasModel with Theano backend) (#46)

* fixes shape error in KerasModel #45 * removes changes in test * fixed flake8
bethgelab · Jul 23, 2017 · bca58bc · bca58bc
1 parent acec8d2
commit bca58bc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/foolbox/models/keras.py b/foolbox/models/keras.py
@@ -58,7 +58,7 @@ def __init__(
         elif predicts == 'logits':
             predictions_are_logits = True
 
-        shape = predictions.get_shape().as_list()
+        shape = K.int_shape(predictions)
         _, num_classes = shape
         assert num_classes is not None
 

diff --git a/foolbox/tests/test_models_keras.py b/foolbox/tests/test_models_keras.py
@@ -50,7 +50,6 @@ def test_keras_model(num_classes):
 
 @pytest.mark.parametrize('num_classes', [10, 1000])
 def test_keras_model_probs(num_classes):
-
     bounds = (0, 255)
     channels = num_classes