fixed readme example (#43)

bethgelab · Jul 22, 2017 · 60dd526 · 60dd526
1 parent 70b4c31
commit 60dd526
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.rst b/README.rst
@@ -38,12 +38,13 @@ Example
 
    import foolbox
    import keras
+   import numpy
    from keras.applications.resnet50 import ResNet50
 
    # instantiate model
    keras.backend.set_learning_phase(0)
    kmodel = ResNet50(weights='imagenet')
-   preprocessing = (np.array([104, 116, 123])[None, None], 1)
+   preprocessing = (numpy.array([104, 116, 123]), 1)
    fmodel = foolbox.models.KerasModel(kmodel, bounds=(0, 255), preprocessing=preprocessing)
 
    # get source image and label