Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
bcf6cda1
Unverified
Commit
bcf6cda1
authored
Jun 06, 2024
by
Nicolas Hug
Committed by
GitHub
Jun 06, 2024
Browse files
SBDataset: Only download noval file when image_set='train_noval' (#8475)
parent
a4713054
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
torchvision/datasets/sbd.py
torchvision/datasets/sbd.py
+3
-1
No files found.
torchvision/datasets/sbd.py
View file @
bcf6cda1
...
...
@@ -81,7 +81,9 @@ class SBDataset(VisionDataset):
for
f
in
[
"cls"
,
"img"
,
"inst"
,
"train.txt"
,
"val.txt"
]:
old_path
=
os
.
path
.
join
(
extracted_ds_root
,
f
)
shutil
.
move
(
old_path
,
sbd_root
)
download_url
(
self
.
voc_train_url
,
sbd_root
,
self
.
voc_split_filename
,
self
.
voc_split_md5
)
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
)
if
not
os
.
path
.
isdir
(
sbd_root
):
raise
RuntimeError
(
"Dataset not found or corrupted. You can use download=True to download it"
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment