From 826706de354c6929c4ee2f9d8c497e12e8257233 Mon Sep 17 00:00:00 2001 From: chenguowei01 Date: Thu, 6 Aug 2020 11:00:02 +0800 Subject: [PATCH] update voc.py --- dygraph/datasets/voc.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dygraph/datasets/voc.py b/dygraph/datasets/voc.py index f0614091..7e2c1d4e 100644 --- a/dygraph/datasets/voc.py +++ b/dygraph/datasets/voc.py @@ -100,5 +100,6 @@ class PascalVOC(Dataset): for line in f: line = line.strip() image_path = os.path.join(img_dir, ''.join([line, '.jpg'])) - grt_path = os.path.join(grt_dir, ''.join([line, '.png'])) + grt_path = os.path.join(grt_dir_aug, ''.join([line, + '.png'])) self.file_list.append([image_path, grt_path]) -- GitLab