Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Add .tar support for datasets (#8963)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-08-14 23:06:42 +02:00
1 parent 3fc89994e9
commit d7bc5d7773
1 file changed
+3 -1
Download Patch File Download Diff File Expand all files Collapse all files
utils/general.py
+3 -1
View File
@@ -594,10 +594,12 @@ def download(url, dir='.', unzip=True, delete=True, curl=False, threads=1, retry
else: else:
LOGGER.warning(f'Failed to download {url}...') 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}...') LOGGER.info(f'Unzipping {f}...')
if f.suffix == '.zip': if f.suffix == '.zip':
ZipFile(f).extractall(path=dir) # unzip ZipFile(f).extractall(path=dir) # unzip
elif f.suffix == '.tar':
os.system(f'tar xf {f} --directory {f.parent}') # unzip
elif f.suffix == '.gz': elif f.suffix == '.gz':
os.system(f'tar xfz {f} --directory {f.parent}') # unzip os.system(f'tar xfz {f} --directory {f.parent}') # unzip
if delete: if delete: