123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- import numpy as np
- import cv2
- from math import sqrt
- _LINE_THICKNESS_SCALING = 500.0
- np.random.seed(0)
- RAND_COLORS = np.random.randint(50, 255, (64, 3), "int") # used for class visu
- RAND_COLORS[0] = [220, 220, 220]
- def render_box(img, box, color=(200, 200, 200)):
- """
- Render a box. Calculates scaling and thickness automatically.
- :param img: image to render into
- :param box: (x1, y1, x2, y2) - box coordinates
- :param color: (b, g, r) - box color
- :return: updated image
- """
- x1, y1, x2, y2 = box
- thickness = int(
- round(
- (img.shape[0] * img.shape[1])
- / (_LINE_THICKNESS_SCALING * _LINE_THICKNESS_SCALING)
- )
- )
- thickness = max(1, thickness)
- img = cv2.rectangle(
- img,
- (int(x1), int(y1)),
- (int(x2), int(y2)),
- color,
- thickness=thickness
- )
- return img
- def render_filled_box(img, box, color=(200, 200, 200)):
- """
- Render a box. Calculates scaling and thickness automatically.
- :param img: image to render into
- :param box: (x1, y1, x2, y2) - box coordinates
- :param color: (b, g, r) - box color
- :return: updated image
- """
- x1, y1, x2, y2 = box
- img = cv2.rectangle(
- img,
- (int(x1), int(y1)),
- (int(x2), int(y2)),
- color,
- thickness=cv2.FILLED
- )
- return img
- _TEXT_THICKNESS_SCALING = 700.0
- _TEXT_SCALING = 520.0
- def get_text_size(img, text, normalised_scaling=1.0):
- """
- Get calculated text size (as box width and height)
- :param img: image reference, used to determine appropriate text scaling
- :param text: text to display
- :param normalised_scaling: additional normalised scaling. Default 1.0.
- :return: (width, height) - width and height of text box
- """
- thickness = int(
- round(
- (img.shape[0] * img.shape[1])
- / (_TEXT_THICKNESS_SCALING * _TEXT_THICKNESS_SCALING)
- )
- * normalised_scaling
- )
- thickness = max(1, thickness)
- scaling = img.shape[0] / _TEXT_SCALING * normalised_scaling
- return cv2.getTextSize(text, cv2.FONT_HERSHEY_SIMPLEX, scaling, thickness)[0]
- def render_text(img, text, pos, color=(200, 200, 200), normalised_scaling=1.0):
- """
- Render a text into the image. Calculates scaling and thickness automatically.
- :param img: image to render into
- :param text: text to display
- :param pos: (x, y) - upper left coordinates of render position
- :param color: (b, g, r) - text color
- :param normalised_scaling: additional normalised scaling. Default 1.0.
- :return: updated image
- """
- x, y = pos
- thickness = int(
- round(
- (img.shape[0] * img.shape[1])
- / (_TEXT_THICKNESS_SCALING * _TEXT_THICKNESS_SCALING)
- )
- * normalised_scaling
- )
- thickness = max(1, thickness)
- scaling = img.shape[0] / _TEXT_SCALING * normalised_scaling
- size = get_text_size(img, text, normalised_scaling)
- cv2.putText(
- img,
- text,
- (int(x), int(y + size[1])),
- cv2.FONT_HERSHEY_SIMPLEX,
- scaling,
- color,
- thickness=thickness,
- )
- return img
|