From fc62d2232d54c1e812115abe376d573e03db4119 Mon Sep 17 00:00:00 2001 From: Baek JeongHun Date: Sat, 10 Aug 2019 18:29:40 +0900 Subject: [PATCH] 'minor' --- dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataset.py b/dataset.py index 515951b..a5a8e32 100755 --- a/dataset.py +++ b/dataset.py @@ -90,7 +90,7 @@ def hierarchical_dataset(root, opt, select_data='/'): """ select_data='/' contains all sub-directory of root directory """ dataset_list = [] print(f'dataset_root: {root}\t dataset: {select_data[0]}') - for dirpath, dirnames, filenames in os.walk(root): + for dirpath, dirnames, filenames in os.walk(root+'/'): if not dirnames: select_flag = False for selected_d in select_data: