mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Annotator `check_font()` RANK -1 remove progress () 

* Annotator `check_font()` RANK -1 remove progress

* Cleanup
pull/4866/head
Glenn Jocher 2021-09-20 13:57:23 +02:00 committed by GitHub
parent 2b3109eeb0
commit 8ad9e4ed5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions

9
utils/plots.py
View File

@ -3,11 +3,12 @@
Plotting utils Plotting utils
""" """
import math
import os
from copy import copy from copy import copy
from pathlib import Path from pathlib import Path
import cv2 import cv2
import math
import matplotlib import matplotlib
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
@ -21,6 +22,7 @@ from utils.metrics import fitness
# Settings # Settings
CONFIG_DIR = user_config_dir() # Ultralytics settings dir CONFIG_DIR = user_config_dir() # Ultralytics settings dir
RANK = int(os.getenv('RANK', -1))
matplotlib.rc('font', **{'size': 11}) matplotlib.rc('font', **{'size': 11})
matplotlib.use('Agg') # for writing to files only matplotlib.use('Agg') # for writing to files only
@ -55,12 +57,13 @@ def check_font(font='Arial.ttf', size=10):
except Exception as e: # download if missing except Exception as e: # download if missing
url = "https://ultralytics.com/assets/" + font.name url = "https://ultralytics.com/assets/" + font.name
print(f'Downloading {url} to {font}...') 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) return ImageFont.truetype(str(font), size)
class Annotator: 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 # 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):