Shuffle fetched images

DeepLabCut · Jun 5, 2020 · d915c2c · d915c2c
1 parent 899f650
commit d915c2c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/model.py b/model.py
@@ -1,9 +1,9 @@
-from skimage import io
-import time
-
+import custom_types
 import pandas as pd
+import random
+import time
+from skimage import io
 
-import custom_types
 
 
 class User:
@@ -24,7 +24,8 @@ class Dataset:
 
     def __init__(self, fnames):
         self.fnames = fnames
-        self.data = io.imread_collection(fnames)
+        random.shuffle(self.fnames)
+        self.data = io.imread_collection(self.fnames)
 
     def __getitem__(self, index):
         if index >= len(self):