Skip to content

Commit

Permalink
Annotator check_font() RANK -1 remove progress (#4864)
Browse files Browse the repository at this point in the history
* Annotator `check_font()` RANK -1 remove progress

* Cleanup
  • Loading branch information
glenn-jocher committed Sep 20, 2021
1 parent 2b3109e commit 8ad9e4e
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions utils/plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,12 @@
Plotting utils
"""

import math
import os
from copy import copy
from pathlib import Path

import cv2
import math
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
Expand All @@ -21,6 +22,7 @@

# Settings
CONFIG_DIR = user_config_dir() # Ultralytics settings dir
RANK = int(os.getenv('RANK', -1))
matplotlib.rc('font', **{'size': 11})
matplotlib.use('Agg') # for writing to files only

Expand Down Expand Up @@ -55,12 +57,13 @@ def check_font(font='Arial.ttf', size=10):
except Exception as e: # download if missing
url = "https://ultralytics.com/assets/" + font.name
print(f'Downloading {url} to {font}...')
torch.hub.download_url_to_file(url, str(font))
torch.hub.download_url_to_file(url, str(font), progress=False)
return ImageFont.truetype(str(font), size)


class Annotator:
check_font() # download TTF if necessary
if RANK in (-1, 0):
check_font() # download TTF if necessary

# 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):
Expand Down

0 comments on commit 8ad9e4e

Please sign in to comment.