Kaydet (Commit) 0bab4b2b authored tarafından bw4sz's avatar bw4sz

first make groups, then randomize on init

üst 3530bc0f
......@@ -79,13 +79,13 @@ class Generator(keras.utils.Sequence):
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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment