Add .tar support for datasets (#8963)

ultralytics · Aug 14, 2022 · d7bc5d7 · d7bc5d7
1 parent 3fc8999
commit d7bc5d7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/utils/general.py b/utils/general.py
@@ -594,10 +594,12 @@ def download_one(url, dir):
                 else:
                     LOGGER.warning(f'Failed to download {url}...')
 
-        if unzip and success and f.suffix in ('.zip', '.gz'):
+        if unzip and success and f.suffix in ('.zip', '.tar', '.gz'):
             LOGGER.info(f'Unzipping {f}...')
             if f.suffix == '.zip':
                 ZipFile(f).extractall(path=dir)  # unzip
+            elif f.suffix == '.tar':
+                os.system(f'tar xf {f} --directory {f.parent}')  # unzip
             elif f.suffix == '.gz':
                 os.system(f'tar xfz {f} --directory {f.parent}')  # unzip
             if delete: