mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Auto-download Arial.ttf on init (#4606)

Browse Source 
* Auto-download Arial.ttf on init

* Fix ROOT
This commit is contained in:
Glenn Jocher 2021-08-30 14:58:22 +02:00 committed by GitHub
parent bb4da083d1
commit e7d1842a98
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
utils/__init__.py
View File

@ -0,0 +1,16 @@
from pathlib import Path
import torch
from PIL import ImageFont
FILE = Path(__file__).absolute()
ROOT = FILE.parents[1] # yolov5/ dir
# Check YOLOv5 Annotator font
font = 'Arial.ttf'
try:
ImageFont.truetype(font)
except Exception as e: # download if missing
url = "https://ultralytics.com/assets/" + font
print(f'Downloading {url} to {ROOT / font}...')
torch.hub.download_url_to_file(url, str(ROOT / font))

9
utils/plots.py
View File

@ -48,7 +48,7 @@ colors = Colors() # create instance for 'from utils.plots import colors'
class Annotator: class Annotator:
# YOLOv5 Annotator for train/val mosaics and jpgs and detect/hub inference annotations # YOLOv5 Annotator for train/val mosaics and jpgs and detect/hub inference annotations
def __init__(self, im, line_width=None, font_size=None, font='Arial.ttf', pil=True): def __init__(self, im, line_width=None, font_size=None, font='Arial.ttf', pil=True):
assert im.data.contiguous, 'Image not contiguous. Apply np.ascontiguousarray(im) to plot_on_box() input image.' assert im.data.contiguous, 'Image not contiguous. Apply np.ascontiguousarray(im) to Annotator() input images.'
self.pil = pil self.pil = pil
if self.pil: # use PIL if self.pil: # use PIL
self.im = im if isinstance(im, Image.Image) else Image.fromarray(im) self.im = im if isinstance(im, Image.Image) else Image.fromarray(im)
@ -57,11 +57,10 @@ class Annotator:
f = font_size or max(round(s * 0.035), 12) f = font_size or max(round(s * 0.035), 12)
try: try:
self.font = ImageFont.truetype(font, size=f) self.font = ImageFont.truetype(font, size=f)
except Exception as e: # download TTF if missing except Exception as e: # download if missing
print(f'WARNING: Annotator font {font} not found: {e}') url = "https://ultralytics.com/assets/" + font
url = "https://github.com/ultralytics/yolov5/releases/download/v1.0/" + font print(f'Downloading {url} to {font}...')
torch.hub.download_url_to_file(url, font) torch.hub.download_url_to_file(url, font)
print(f'Annotator font successfully downloaded from {url} to {font}')
self.font = ImageFont.truetype(font, size=f) self.font = ImageFont.truetype(font, size=f)
self.fh = self.font.getsize('a')[1] - 3 # font height self.fh = self.font.getsize('a')[1] - 3 # font height
else: # use cv2 else: # use cv2