Skip to content

GitLab

Unverified Commit bcf6cda1 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

SBDataset: Only download noval file when image_set='train_noval' (#8475)

parent a4713054
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
torchvision/datasets/sbd.py
View file @ bcf6cda1
...@@ -81,6 +81,8 @@ class SBDataset(VisionDataset): ...@@ -81,6 +81,8 @@ class SBDataset(VisionDataset):
for f in ["cls", "img", "inst", "train.txt", "val.txt"]: for f in ["cls", "img", "inst", "train.txt", "val.txt"]:
old_path = os.path.join(extracted_ds_root, f) old_path = os.path.join(extracted_ds_root, f)
shutil.move(old_path, sbd_root) shutil.move(old_path, sbd_root)
if self.image_set == "train_noval":
# Note: this is failing as of June 2024 https://github.com/pytorch/vision/issues/8471
download_url(self.voc_train_url, sbd_root, self.voc_split_filename, self.voc_split_md5) download_url(self.voc_train_url, sbd_root, self.voc_split_filename, self.voc_split_md5)
if not os.path.isdir(sbd_root): if not os.path.isdir(sbd_root):
......
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