check_fonts() download to CONFIG_DIR fix (#7489)

Follows https://github.com/ultralytics/yolov5/pull/7488. Correct bug where fonts were downloading to current working directory rather than global CONFIG_DIR
This commit is contained in:
Glenn Jocher 2022-04-19 17:50:02 -07:00 committed by GitHub
parent c9042dc2ad
commit ab5b917494
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -427,10 +427,11 @@ def check_file(file, suffix=''):
def check_font(font=FONT, progress=False):
# Download font to CONFIG_DIR if necessary
font = Path(font)
if not font.exists() and not (CONFIG_DIR / font.name).exists():
file = CONFIG_DIR / font.name
if not font.exists() and not file.exists():
url = "https://ultralytics.com/assets/" + font.name
LOGGER.info(f'Downloading {url} to {CONFIG_DIR / font.name}...')
torch.hub.download_url_to_file(url, str(font), progress=progress)
LOGGER.info(f'Downloading {url} to {file}...')
torch.hub.download_url_to_file(url, str(file), progress=progress)
def check_dataset(data, autodownload=True):