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

Add unzip flag to download() (#3002)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-05-01 12:37:41 +02:00
1 parent f55730ee9c
commit 4d7bca7648
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
utils/general.py
+2 -2
View File
@@ -183,7 +183,7 @@ def check_dataset(dict):
raise Exception('Dataset not found.') raise Exception('Dataset not found.')
def download(url, dir='.', threads=1): def download(url, dir='.', unzip=True, threads=1):
# Multi-threaded file download and unzip function # Multi-threaded file download and unzip function
def download_one(url, dir): def download_one(url, dir):
# Download 1 file # Download 1 file
@@ -191,7 +191,7 @@ def download(url, dir='.', threads=1):
if not f.exists(): if not f.exists():
print(f'Downloading {url} to {f}...') print(f'Downloading {url} to {f}...')
torch.hub.download_url_to_file(url, f, progress=True) # download torch.hub.download_url_to_file(url, f, progress=True) # download
if f.suffix in ('.zip', '.gz'): if unzip and f.suffix in ('.zip', '.gz'):
print(f'Unzipping {f}...') print(f'Unzipping {f}...')
if f.suffix == '.zip': if f.suffix == '.zip':
os.system(f'unzip -qo {f} -d {dir} && rm {f}') # unzip -quiet -overwrite os.system(f'unzip -qo {f} -d {dir} && rm {f}') # unzip -quiet -overwrite