Fixed shuffling at end of epoch when shuffling is False

huynhtnhut97 · Jan 16, 2019 · 9db1b0a · 9db1b0a
1 parent a619206
commit 9db1b0a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/keras_retinanet/preprocessing/generator.py b/keras_retinanet/preprocessing/generator.py
@@ -88,7 +88,8 @@ def __init__(
             self.on_epoch_end()
 
     def on_epoch_end(self):
-        random.shuffle(self.groups)
+        if self.shuffle_groups:
+            random.shuffle(self.groups)
 
     def size(self):
         """ Size of the dataset.