|
import cv2 |
|
|
|
|
|
def draw_text(img, text, |
|
font=cv2.FONT_HERSHEY_PLAIN, |
|
x=0, |
|
y=0, |
|
font_scale=1, |
|
font_thickness=2, |
|
text_color=(0, 255, 0), |
|
text_color_bg=(0, 0, 0) |
|
): |
|
|
|
text_size, _ = cv2.getTextSize(text, font, font_scale, font_thickness) |
|
text_w, text_h = text_size |
|
cv2.rectangle(img, (x, y), (x + text_w, y + text_h), text_color_bg, -1) |
|
cv2.putText(img, text, (x, y + text_h + font_scale - 1), font, font_scale, text_color, font_thickness) |
|
|
|
return img |
|
|
|
|
|
def add_bboxes(img, bboxes, **kwargs): |
|
label = "dog" |
|
font_thickness=2 |
|
for idx, box in enumerate(bboxes): |
|
x1, y1 = int(box[0]), int(box[1]) |
|
x2, y2 = int(box[2]), int(box[3]) |
|
if "scores" in kwargs.keys(): |
|
score = list(kwargs["scores"])[idx] |
|
label = "{0:.2f}%".format(score*100) |
|
font_thickness=1 |
|
img = cv2.rectangle(img, (x1, y1), (x2, y2), (36,255,12), 2) |
|
img = draw_text(img, label, x=x1, y=y1-10, font_thickness=font_thickness) |
|
return img |
|
|
|
|
|
def reformat_bbox(coord): |
|
""" |
|
Converts XYWH to XYXY |
|
""" |
|
x1, y1 = coord[0], coord[1] |
|
x2, y2 = x1 + coord[2], y1 + coord[3] |
|
return [x1, y1, x2, y2] |
|
|
