xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix GlobalWheat2020.yaml auto download (#3007)

Browse Source
This commit is contained in:
Glenn Jocher 2023-06-05 01:37:36 +02:00 committed by GitHub
parent 2193ca1658
commit 2b26572e42
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ultralytics/datasets/GlobalWheat2020.yaml
View File

@ -48,7 +48,7 @@ download: |
# Move # Move
for p in 'arvalis_1', 'arvalis_2', 'arvalis_3', 'ethz_1', 'rres_1', 'inrae_1', 'usask_1', \ for p in 'arvalis_1', 'arvalis_2', 'arvalis_3', 'ethz_1', 'rres_1', 'inrae_1', 'usask_1', \
'utokyo_1', 'utokyo_2', 'nau_1', 'uq_1': 'utokyo_1', 'utokyo_2', 'nau_1', 'uq_1':
(dir / p).rename(dir / 'images' / p) # move to /images (dir / 'global-wheat-codalab-official' / p).rename(dir / 'images' / p) # move to /images
f = (dir / p).with_suffix('.json') # json file f = (dir / 'global-wheat-codalab-official' / p).with_suffix('.json') # json file
if f.exists(): if f.exists():
f.rename((dir / 'annotations' / p).with_suffix('.json')) # move to /annotations f.rename((dir / 'annotations' / p).with_suffix('.json')) # move to /annotations

5
ultralytics/yolo/utils/downloads.py
View File

@ -131,9 +131,10 @@ def safe_download(url,
a successful download. Default: 1E0. a successful download. Default: 1E0.
progress (bool, optional): Whether to display a progress bar during the download. Default: True. progress (bool, optional): Whether to display a progress bar during the download. Default: True.
""" """
if '://' not in str(url) and Path(url).is_file(): # exists ('://' check required in Windows Python<3.10) f = dir / url2file(url) if dir else Path(file) # URL converted to filename
if '://' not in str(url) and Path(url).is_file(): # URL exists ('://' check required in Windows Python<3.10)
f = Path(url) # filename f = Path(url) # filename
else: # does not exist elif not f.is_file(): # URL and file do not exist
assert dir or file, 'dir or file required for download' assert dir or file, 'dir or file required for download'
f = dir / url2file(url) if dir else Path(file) f = dir / url2file(url) if dir else Path(file)
desc = f'Downloading {clean_url(url)} to {f}' desc = f'Downloading {clean_url(url)} to {f}'