dir_name

JaminFong · Jun 18, 2020 · 9e5b9d6 · 9e5b9d6
1 parent 883bc04
commit 9e5b9d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataset/mk_split_img_list.py b/dataset/mk_split_img_list.py
@@ -5,7 +5,7 @@
 def get_list(data_path, output_path):
     for split in os.listdir(data_path):
         if split == 'train':
-            split_path = os.path.join(data_path, 'train_img')
+            split_path = os.path.join(data_path, split)
             if not os.path.isdir(split_path):
                 continue
             f_train = open(os.path.join(output_path, split + '_datalist'), 'w')