Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
for label_i in unique_labels:
mask = label == label_i
if 1.0 * mask.sum() / mask.size < thresh_suppress:
continue
y, x = np.array(_center_of_mass(mask), dtype=int)
if label[y, x] != label_i:
Y, X = np.where(mask)
point_index = np.random.randint(0, len(Y))
y, x = Y[point_index], X[point_index]
text = label_names[label_i]
height, width = draw_module.text_size(
text, size=font_size, font_path=font_path
)
color = color_module.get_fg_color(res[y, x])
res = draw_module.text(
res,
yx=(y - height // 2, x - width // 2),
text=text,
color=color,
size=font_size,
font_path=font_path,
)
elif loc in ["rb", "lt"]:
text_sizes = np.array(
[
draw_module.text_size(
label_names[l], font_size, font_path=font_path
)
for l in unique_labels
]
color: (3,) array-like
Text RGB color in uint8.
If None, the color is determined by background color.
(default: None)
aabb1, aabb2: (2,) array-like
Coordinate of the rectangle (y_min, x_min), (y_max, x_max).
Default is (0, 0), (height, width).
Returns
-------
dst: numpy.ndarray
Output image.
"""
if color is None:
color = color_module.get_fg_color(background)
height, width = src.shape[:2]
y1, x1, y2, x2 = text_in_rectangle_aabb(
src=src,
loc=loc,
text=text,
size=size,
aabb1=aabb1,
aabb2=aabb2,
font_path=font_path,
)
constant_values = (
(background[0],),
(background[1],),
(background[2],),