Skip to content

Commit

Permalink
Improve plots.py robustness
Browse files Browse the repository at this point in the history
Addresses issues #5374, #5395, #5611
  • Loading branch information
glenn-jocher committed Nov 11, 2021
1 parent 6903251 commit fde5881
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions utils/plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,8 @@
import torch
from PIL import Image, ImageDraw, ImageFont

from utils.general import clip_coords, increment_path, is_ascii, is_chinese, user_config_dir, xywh2xyxy, xyxy2xywh
from utils.general import clip_coords, increment_path, is_ascii, is_chinese, user_config_dir, xywh2xyxy, xyxy2xywh, \
check_requirements, try_except, Timeout
from utils.metrics import fitness

# Settings
Expand Down Expand Up @@ -58,7 +59,10 @@ def check_font(font='Arial.ttf', size=10):
url = "https://ultralytics.com/assets/" + font.name
print(f'Downloading {url} to {font}...')
torch.hub.download_url_to_file(url, str(font), progress=False)
return ImageFont.truetype(str(font), size)
try:
return ImageFont.truetype(str(font), size)
except TypeError:
check_requirements('Pillow>=8.4.0') # known issue https://github.com/ultralytics/yolov5/issues/5374


class Annotator:
Expand Down Expand Up @@ -320,6 +324,8 @@ def plot_val_study(file='', dir='', x=None): # from utils.plots import *; plot_
plt.savefig(f, dpi=300)


@try_except # known issue https://github.com/ultralytics/yolov5/issues/5395
@Timeout(30) # known issue https://github.com/ultralytics/yolov5/issues/5611
def plot_labels(labels, names=(), save_dir=Path('')):
# plot dataset labels
print('Plotting labels... ')
Expand Down

0 comments on commit fde5881

Please sign in to comment.