Skip to content

Commit d915c2c

Browse files
committed
Shuffle fetched images
1 parent 899f650 commit d915c2c

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

model.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
from skimage import io
2-
import time
3-
1+
import custom_types
42
import pandas as pd
3+
import random
4+
import time
5+
from skimage import io
56

6-
import custom_types
77

88

99
class User:
@@ -24,7 +24,8 @@ class Dataset:
2424

2525
def __init__(self, fnames):
2626
self.fnames = fnames
27-
self.data = io.imread_collection(fnames)
27+
random.shuffle(self.fnames)
28+
self.data = io.imread_collection(self.fnames)
2829

2930
def __getitem__(self, index):
3031
if index >= len(self):

0 commit comments

Comments
 (0)