| 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") |
| 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 |
|
|
