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

Merge remote-tracking branch 'origin/feature/redesigned_annotators' into feature/redesigned_annotators

This commit is contained in:
SkalskiP 2023-09-28 18:08:45 +02:00
parent f7bf766a60 1e0f18f792
commit 507e000bca
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
supervision/annotators/core.py
View File

@ -291,22 +291,17 @@ class BoxCornerAnnotator(BaseAnnotator):
color_map=self.color_map,
)
color = resolve_color(color=self.color, idx=idx)
corners = [
(x1, y1), (x2, y1),
(x1, y2), (x2, y2)
]
corners = [(x1, y1), (x2, y1), (x1, y2), (x2, y2)]
for x, y in corners:
x_end = x + self.corner_length if x == x1 else x - self.corner_length
cv2.line(
scene, (x, y), (x_end, y),
color.as_bgr(), thickness=self.thickness
scene, (x, y), (x_end, y), color.as_bgr(), thickness=self.thickness
)
y_end = y + self.corner_length if y == y1 else y - self.corner_length
cv2.line(
scene, (x, y), (x, y_end),
color.as_bgr(), thickness=self.thickness
scene, (x, y), (x, y_end), color.as_bgr(), thickness=self.thickness
)
return scene