diff --git a/utils/general.py b/utils/general.py index 8469773f17f2..10fa07f379f3 100755 --- a/utils/general.py +++ b/utils/general.py @@ -480,7 +480,7 @@ def check_dataset(data, autodownload=True): # Download (optional) extract_dir = '' if isinstance(data, (str, Path)) and str(data).endswith('.zip'): # i.e. gs://bucket/dir/coco128.zip - download(data, dir=DATASETS_DIR, unzip=True, delete=False, curl=False, threads=1) + download(data, dir=f'{DATASETS_DIR}/{Path(data).stem}', unzip=True, delete=False, curl=False, threads=1) data = next((DATASETS_DIR / Path(data).stem).rglob('*.yaml')) extract_dir, autodownload = data.parent, False