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

Updated the Triangle_annotator and the Dot_annotator in the annotator-core.py file

This commit is contained in:
Dripto Saha 2024-06-20 00:12:42 +05:30 committed by GitHub
parent d821fca2af
commit 0d0328ef22
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
supervision/annotators/core.py
View File

@ -901,8 +901,13 @@ class DotAnnotator(BaseAnnotator):
else custom_color_lookup,
)
center = (int(xy[detection_idx, 0]), int(xy[detection_idx, 1]))
cv2.circle(scene, center, self.radius, (0, 0, 0), self.outline_thickness)
cv2.circle(scene, center, self.radius, color.as_bgr(), -1)
if self.outline_thickness:
cv2.circle(scene, center, self.radius, (0,0,0), self.outline_thickness)
cv2.circle(scene, center, self.radius, color.as_bgr(), -1)
else:
cv2.circle(scene, center, self.radius, color.as_bgr(), -1)
return scene