|
import cv2 |
|
import numpy as np |
|
|
|
def draw_bbox(image,box,color=(255,0,0),thickness=1): |
|
if thickness==0: |
|
return |
|
|
|
left = int(box[0]) |
|
top = int(box[1]) |
|
right = int(box[0]+box[2]) |
|
bottom = int(box[1]+box[3]) |
|
box_points =[(left,top),(right,top),(right,bottom),(left,bottom)] |
|
|
|
cv2.polylines(image, [np.array(box_points)], isClosed=True, color=color, thickness=thickness) |
|
|
|
|
|
def to_int_points(points): |
|
int_points=[] |
|
for point in points: |
|
int_points.append([int(point[0]),int(point[1])]) |
|
return int_points |
|
|
|
def draw_text(img, text, point, font_scale=0.5, color=(200, 200, 200), thickness=1): |
|
font = cv2.FONT_HERSHEY_SIMPLEX |
|
cv2.putText(img, str(text), point, font, font_scale, color, thickness, cv2.LINE_AA) |
|
|
|
plot_text_color = (200, 200, 200) |
|
plot_text_font_scale = 0.5 |
|
plot_index = 1 |
|
plot_text = True |
|
|
|
def set_plot_text(is_plot,text_font_scale,text_color): |
|
global plot_index,plot_text,plot_text_font_scale,plot_text_color |
|
plot_text = is_plot |
|
plot_index = 1 |
|
plot_text_font_scale = text_font_scale |
|
plot_text_color = text_color |
|
|
|
def plot_points(image,points,isClosed=False,circle_size=3,circle_color=(255,0,0),line_size=1,line_color=(0,0,255)): |
|
global plot_index,plot_text |
|
int_points = to_int_points(points) |
|
if circle_size>0: |
|
for point in int_points: |
|
cv2.circle(image,point,circle_size,circle_color,-1) |
|
if plot_text: |
|
draw_text(image,plot_index,point,plot_text_font_scale,plot_text_color) |
|
plot_index+=1 |
|
if line_size>0: |
|
cv2.polylines(image, [np.array(int_points)], isClosed=isClosed, color=line_color, thickness=line_size) |