From 15516422d31f7562a03ec95497f5a1cce3b6c8fe Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 24 Apr 2021 15:53:15 +0200 Subject: [PATCH] Update download() for tar.gz files (#2919) * Update download() for tar.gz files * Update general.py --- utils/general.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/utils/general.py b/utils/general.py index ba88759c2983..8efeb5ea59cc 100755 --- a/utils/general.py +++ b/utils/general.py @@ -184,14 +184,19 @@ def check_dataset(dict): def download(url, dir='.', multi_thread=False): - # Multi-threaded file download function + # Multi-threaded file download and unzip function def download_one(url, dir): # Download 1 file f = dir / Path(url).name # filename - print(f'Downloading {url} to {f}...') - torch.hub.download_url_to_file(url, f, progress=True) # download - if f.suffix == '.zip': - os.system(f'unzip -qo {f} -d {dir} && rm {f}') # unzip -quiet -overwrite + if not f.exists(): + print(f'Downloading {url} to {f}...') + torch.hub.download_url_to_file(url, f, progress=True) # download + if f.suffix in ('.zip', '.gz'): + print(f'Unzipping {f}...') + if f.suffix == '.zip': + os.system(f'unzip -qo {f} -d {dir} && rm {f}') # unzip -quiet -overwrite + elif f.suffix == '.gz': + os.system(f'tar xfz {f} --directory {f.parent} && rm {f}') # unzip dir = Path(dir) dir.mkdir(parents=True, exist_ok=True) # make directory