From e8f6b3f1aad7616acb8157cae4f994dc15063020 Mon Sep 17 00:00:00 2001 From: tingsong Date: Tue, 27 Aug 2019 19:25:54 +0800 Subject: [PATCH] fix bug --- Code/1_data_prepare/1_2_split_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Code/1_data_prepare/1_2_split_dataset.py b/Code/1_data_prepare/1_2_split_dataset.py index da01167..1e20fb6 100644 --- 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) -- GitLab