first make groups, then randomize on init

huynhtnhut97 · Dec 6, 2018 · 0bab4b2 · 0bab4b2
1 parent 3530bc0
commit 0bab4b2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/keras_retinanet/preprocessing/generator.py b/keras_retinanet/preprocessing/generator.py
@@ -79,13 +79,13 @@ def __init__(
         self.preprocess_image       = preprocess_image
         self.config                 = config
 
+        #Define groups
+        self.group_images()
+
         #Shuffle when initializing 
         if self.shuffle_groups:
             self.on_epoch_end()
 
-        #Define groups
-        self.group_images()
-
     def on_epoch_end(self):
         random.shuffle(self.groups)