Explicitly use the Theano image ordering (in case of using the Tensor…

…Flow backend).
chriii · May 29, 2017 · 9f2fca9 · 9f2fca9
1 parent 84b917d
commit 9f2fca9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/deeplearning1/nbs/vgg16.py b/deeplearning1/nbs/vgg16.py
@@ -16,6 +16,10 @@
 from keras.optimizers import SGD, RMSprop, Adam
 from keras.preprocessing import image
 
+# In case we are going to use the TensorFlow backend we need to explicitly set the Theano image ordering
+from keras import backend as K
+K.set_image_dim_ordering('th')
+
 
 vgg_mean = np.array([123.68, 116.779, 103.939], dtype=np.float32).reshape((3,1,1))
 def vgg_preprocess(x):

diff --git a/deeplearning1/nbs/vgg16bn.py b/deeplearning1/nbs/vgg16bn.py
@@ -16,6 +16,9 @@
 from keras.optimizers import SGD, Adam
 from keras.preprocessing import image
 
+# In case we are going to use the TensorFlow backend we need to explicitly set the Theano image ordering
+from keras import backend as K
+K.set_image_dim_ordering('th')
 
 vgg_mean = np.array([123.68, 116.779, 103.939], dtype=np.float32).reshape((3,1,1))
 def vgg_preprocess(x):