Skip to content

GitLab

Unverified Commit c00c4d36 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix VOC datasets for 2007 (#3572) 


Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
parent 00c460af
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
torchvision/datasets/voc.py
View file @ c00c4d36
......@@ -87,10 +87,9 @@ class _VOCBase(VisionDataset):
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"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment