fix issue with unsorted file names

YangLei678 · Nov 16, 2018 · 8dffbca · 8dffbca
1 parent 713b225
commit 8dffbca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hw6_release/utils.py b/hw6_release/utils.py
@@ -44,7 +44,7 @@ def load_dataset(data_dir, train=True, as_grey=False, shuffle=True, seed=131):
         data_dir = os.path.join(data_dir, 'test')
 
     for i, cls in enumerate(sorted(os.listdir(data_dir))):
-        for img_file in os.listdir(os.path.join(data_dir, cls)):
+        for img_file in sorted(os.listdir(os.path.join(data_dir, cls))):
             img_path = os.path.join(data_dir, cls, img_file)
             img = img_as_float(io.imread(img_path, as_grey=as_grey))
             X.append(img)