Fix initialization of index_array

index_array should be initialized when self.batch_index is zero.
keras-team · May 7, 2016 · 2601dcd · 2601dcd
1 parent bb626c1
commit 2601dcd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/keras/preprocessing/image.py b/keras/preprocessing/image.py
@@ -248,9 +248,12 @@ def reset(self):
         self.batch_index = 0
 
     def _flow_index(self, N, batch_size=32, shuffle=False, seed=None):
+        # ensure self.batch_index is 0
+        self.reset()
+
         while 1:
-            index_array = np.arange(N)
             if self.batch_index == 0:
+                index_array = np.arange(N)
                 if shuffle:
                     if seed is not None:
                         np.random.seed(seed + self.total_batches_seen)