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

🧪 Test of BoxCornerAnnotator alternative annotate method implementation.

Browse Source
This commit is contained in:
SkalskiP committed 2023-09-28 17:11:06 +02:00
1 parent de4056192e
commit c35a7714d2
1 file changed
+17 -56
Download Patch File Download Diff File Expand all files Collapse all files
supervision/annotators/core.py
+17 -56
View File
@@ -290,62 +290,23 @@ class BoxCornerAnnotator(BaseAnnotator):
color_map=self.color_map,
)
color = resolve_color(color=self.color, idx=idx)
cv2.line(
scene,
(x1, y1),
(x1 + self.corner_length, y1),
color.as_bgr(),
thickness=self.thickness,
)
cv2.line(
scene,
(x2 - self.corner_length, y1),
(x2, y1),
color.as_bgr(),
thickness=self.thickness,
)
cv2.line(
scene,
(x1, y2),
(x1 + self.corner_length, y2),
color.as_bgr(),
thickness=self.thickness,
)
cv2.line(
scene,
(x2 - self.corner_length, y2),
(x2, y2),
color.as_bgr(),
thickness=self.thickness,
)
cv2.line(
scene,
(x1, y1),
(x1, y1 + self.corner_length),
color.as_bgr(),
thickness=self.thickness,
)
cv2.line(
scene,
(x2, y1 + self.corner_length),
(x2, y1),
color.as_bgr(),
thickness=self.thickness,
)
cv2.line(
scene,
(x1, y2 - self.corner_length),
(x1, y2),
color.as_bgr(),
thickness=self.thickness,
)
cv2.line(
scene,
(x2, y2 - self.corner_length),
(x2, y2),
color.as_bgr(),
thickness=self.thickness,
)
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
)
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
)
return scene