From a9553886ad3df03eee7871d398e9635b338f79ac Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Wed, 17 Mar 2021 16:08:29 +0100 Subject: [PATCH] fix VOC datasets for 2007 (#3572) Co-authored-by: Francisco Massa --- torchvision/datasets/voc.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/torchvision/datasets/voc.py b/torchvision/datasets/voc.py index b905ede7a0a..9f6685e865c 100644 --- a/torchvision/datasets/voc.py +++ b/torchvision/datasets/voc.py @@ -87,10 +87,9 @@ def __init__( valid_image_sets = ["train", "trainval", "val"] if year == "2007": valid_image_sets.append("test") - key = "2007-test" - else: - key = year self.image_set = verify_str_arg(image_set, "image_set", valid_image_sets) + + key = "2007-test" if year == "2007" and image_set == "test" else year dataset_year_dict = DATASET_YEAR_DICT[key] self.url = dataset_year_dict["url"]