diff --git a/dygraph/datasets/voc.py b/dygraph/datasets/voc.py index f061409185eabc8a1ade0714fb69e3fa3d199040..7e2c1d4e8d4b157084beb90734660f05c9377c19 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])