From 9e5b9d688081311dbeb5e197673ea3771ef8695f Mon Sep 17 00:00:00 2001 From: "jiemin.fang" Date: Thu, 18 Jun 2020 17:20:08 +0800 Subject: [PATCH] dir_name --- dataset/mk_split_img_list.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataset/mk_split_img_list.py b/dataset/mk_split_img_list.py index 1d471d0..23b0f36 100644 --- 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')