Commit 394de98e authored by Tomas Alori's avatar Tomas Alori Committed by Francisco Massa
Browse files

Fix calls to 'download_and_extract_archive()' (#1006)

Changes made in 7716aba5 broke calls to this method.
parent 8a64dbcd
......@@ -116,13 +116,13 @@ class Caltech101(VisionDataset):
download_and_extract_archive(
"http://www.vision.caltech.edu/Image_Datasets/Caltech101/101_ObjectCategories.tar.gz",
self.root,
"101_ObjectCategories.tar.gz",
"b224c7392d521a49829488ab0f1120d9")
filename="101_ObjectCategories.tar.gz",
md5="b224c7392d521a49829488ab0f1120d9")
download_and_extract_archive(
"http://www.vision.caltech.edu/Image_Datasets/Caltech101/Annotations.tar",
self.root,
"101_Annotations.tar",
"6f83eeb1f24d99cab4eb377263132c91")
filename="101_Annotations.tar",
md5="6f83eeb1f24d99cab4eb377263132c91")
def extra_repr(self):
return "Target type: {target_type}".format(**self.__dict__)
......@@ -204,5 +204,5 @@ class Caltech256(VisionDataset):
download_and_extract_archive(
"http://www.vision.caltech.edu/Image_Datasets/Caltech256/256_ObjectCategories.tar",
self.root,
"256_ObjectCategories.tar",
"67b4f42ca05d46448c6bb8ecd2220f6d")
filename="256_ObjectCategories.tar",
md5="67b4f42ca05d46448c6bb8ecd2220f6d")
......@@ -147,7 +147,7 @@ class CIFAR10(VisionDataset):
if self._check_integrity():
print('Files already downloaded and verified')
return
download_and_extract_archive(self.url, self.root, self.filename, self.tgz_md5)
download_and_extract_archive(self.url, self.root, filename=self.filename, md5=self.tgz_md5)
def extra_repr(self):
return "Split: {}".format("Train" if self.train is True else "Test")
......
......@@ -88,7 +88,7 @@ class Omniglot(VisionDataset):
filename = self._get_target_folder()
zip_filename = filename + '.zip'
url = self.download_url_prefix + '/' + zip_filename
download_and_extract_archive(url, self.root, zip_filename, self.zips_md5[filename])
download_and_extract_archive(url, self.root, filename=zip_filename, md5=self.zips_md5[filename])
def _get_target_folder(self):
return 'images_background' if self.background else 'images_evaluation'
......@@ -152,7 +152,7 @@ class STL10(VisionDataset):
if self._check_integrity():
print('Files already downloaded and verified')
return
download_and_extract_archive(self.url, self.root, self.filename, self.tgz_md5)
download_and_extract_archive(self.url, self.root, filename=self.filename, md5=self.tgz_md5)
def extra_repr(self):
return "Split: {split}".format(**self.__dict__)
......
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