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

Updated dot-annotator-core.py

This commit is contained in:
Dripto Saha 2024-06-19 19:11:52 +05:30 committed by GitHub
parent ff5709def2
commit 6ea07a1c2a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
supervision/annotators/core.py
View File

@ -830,9 +830,10 @@ class DotAnnotator(BaseAnnotator):
def __init__(
self,
color: Union[Color, ColorPalette] = ColorPalette.DEFAULT,
radius: int = 4,
radius: int = 5,
position: Position = Position.CENTER,
color_lookup: ColorLookup = ColorLookup.CLASS,
outer_thickness : int = 0
):
"""
Args:
@ -847,6 +848,7 @@ class DotAnnotator(BaseAnnotator):
self.radius: int = radius
self.position: Position = position
self.color_lookup: ColorLookup = color_lookup
self.outer_thickness = outer_thickness
@convert_for_annotation_method
def annotate(
@ -899,6 +901,7 @@ class DotAnnotator(BaseAnnotator):
)
center = (int(xy[detection_idx, 0]), int(xy[detection_idx, 1]))
cv2.circle(scene, center, self.radius, color.as_bgr(), -1)
cv2.circle(scene, center, self.radius, (0,0,0), self.outer_thickness)
return scene