Skip to content

Commit 4b21a2f

Browse files
committed
Simplify the datasets existence
1 parent a5b701e commit 4b21a2f

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

torchvision/datasets/flowers102.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -91,9 +91,6 @@ def __getitem__(self, idx) -> Tuple[Any, Any]:
9191
def extra_repr(self) -> str:
9292
return f"split={self._split}"
9393

94-
def _check_exists(self) -> bool:
95-
return self._images_folder.exists() and self._images_folder.is_dir()
96-
9794
def _check_integrity(self):
9895
for id in ["label", "setid"]:
9996
filename, md5 = self._file_dict[id]
@@ -102,14 +99,13 @@ def _check_integrity(self):
10299
return True
103100

104101
def download(self):
105-
if not self._check_exists():
106-
download_and_extract_archive(
107-
f"{self._download_url_prefix}{self._file_dict['image'][0]}",
108-
str(self._base_folder),
109-
md5=self._file_dict["image"][1],
110-
)
111102
if self._check_integrity():
112103
return
104+
download_and_extract_archive(
105+
f"{self._download_url_prefix}{self._file_dict['image'][0]}",
106+
str(self._base_folder),
107+
md5=self._file_dict["image"][1],
108+
)
113109
for id in ["label", "setid"]:
114110
filename, md5 = self._file_dict[id]
115111
download_url(self._download_url_prefix + filename, str(self._base_folder), md5=md5)

0 commit comments

Comments
 (0)