xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'VertexLabelAnnotator' of https://github.com/Bhavay-2001/roboflow-supervision into VertexLabelAnnotator

This commit is contained in:
Bhavay Malhotra 2024-08-07 22:39:45 +05:30
parent 4e7f69a502 6d719fcd60
commit a72c5044f4
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
supervision/keypoint/annotators.py
View File

@ -7,7 +7,7 @@ import numpy as np
from supervision import Rect, pad_boxes
from supervision.annotators.base import ImageType
from supervision.draw.color import Color, ColorPalette
from supervision.draw.color import Color
from supervision.draw.utils import draw_rounded_rectangle
from supervision.keypoint.core import KeyPoints
from supervision.keypoint.skeletons import SKELETONS_BY_VERTEX_COUNT
@ -353,7 +353,9 @@ class VertexLabelAnnotator:
xyxy_padded = pad_boxes(xyxy=xyxy, px=self.text_padding)
for text, color, text_color, box, box_padded in zip(labels, colors, text_colors, xyxy, xyxy_padded):
for text, color, text_color, box, box_padded in zip(
labels, colors, text_colors, xyxy, xyxy_padded
):
draw_rounded_rectangle(
scene=scene,
rect=Rect.from_xyxy(box_padded),
@ -442,4 +444,4 @@ class VertexLabelAnnotator:
np.array(text_colors * skeletons_count)
if isinstance(text_colors, list)
else np.array([text_colors] * points_count * skeletons_count)
)
)