Update officecaltech.py

oldschool24 · Aug 8, 2023 · d87fc5d · d87fc5d
1 parent d28dabf
commit d87fc5d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tllib/vision/datasets/officecaltech.py b/tllib/vision/datasets/officecaltech.py
@@ -51,9 +51,10 @@ def __init__(self, root: str, task: str, download: Optional[bool] = False, **kwa
         if download:
             for dir in self.directories.values():
                 if not os.path.exists(os.path.join(root, dir)):
-                    download_and_extract_archive(url="https://cloud.tsinghua.edu.cn/f/e93f2e07d93243d6b57e/?dl=1",
+                    download_and_extract_archive(url="https://cloud.tsinghua.edu.cn/f/eea518fa781a41d1b20e/?dl=1",
                                                  download_root=os.path.join(root, 'download'),
-                                                 filename="officecaltech.tgz", remove_finished=False, extract_root=root)
+                                                 filename="office-caltech.tgz", remove_finished=False,
+                                                 extract_root=root)
                     break
         else:
             list(map(lambda dir, _: check_exits(root, dir), self.directories.values()))
@@ -72,4 +73,4 @@ def num_classes(self):
 
     @classmethod
     def domains(cls):
-        return list(cls.directories.keys())
+        return list(cls.directories.keys())