Skip to content

GitLab

Commit 8bd05e67 authored by Dan Hendrycks's avatar Dan Hendrycks Committed by Soumith Chintala
Browse files

Expand user path for PASCAL VOC (#681)

parent 71182bc1
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/datasets/voc.py
View file @ 8bd05e67
...@@ -74,7 +74,7 @@ class VOCSegmentation(data.Dataset): ...@@ -74,7 +74,7 @@ class VOCSegmentation(data.Dataset):
download=False, download=False,
transform=None, transform=None,
target_transform=None): target_transform=None):
self.root = root self.root = os.path.expanduser(root)
self.year = year self.year = year
self.url = DATASET_YEAR_DICT[year]['url'] self.url = DATASET_YEAR_DICT[year]['url']
self.filename = DATASET_YEAR_DICT[year]['filename'] self.filename = DATASET_YEAR_DICT[year]['filename']
...@@ -157,7 +157,7 @@ class VOCDetection(data.Dataset): ...@@ -157,7 +157,7 @@ class VOCDetection(data.Dataset):
download=False, download=False,
transform=None, transform=None,
target_transform=None): target_transform=None):
self.root = root self.root = os.path.expanduser(root)
self.year = year self.year = year
self.url = DATASET_YEAR_DICT[year]['url'] self.url = DATASET_YEAR_DICT[year]['url']
self.filename = DATASET_YEAR_DICT[year]['filename'] self.filename = DATASET_YEAR_DICT[year]['filename']
......
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