fix bug

e8f6b3f1 · tingsong · 432cf408 · e8f6b3f1
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

Code/1_data_prepare/1_2_split_dataset.py Code/1_data_prepare/1_2_split_dataset.py +1 -1

未找到文件。
--- a/Code/1_data_prepare/1_2_split_dataset.py
+++ b/Code/1_data_prepare/1_2_split_dataset.py
@@ -27,7 +27,7 @@ if __name__ == '__main__':

    for root, dirs, files in os.walk(dataset_dir):
        for sDir in dirs:
-            imgs_list = glob.glob(os.path.join(root, sDir) + '*.png')
+            imgs_list = glob.glob(os.path.join(root, sDir, '*.png'))
            random.seed(666)
            random.shuffle(imgs_list)
            imgs_num = len(imgs_list)