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

fix precommit issue

This commit is contained in:
James Gallagher 2023-11-27 11:15:55 +00:00
parent 9e5d30732d
commit b6c85a3f4b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
supervision/annotators/core.py
View File

@ -984,7 +984,9 @@ class ClassificationAnnotator:
>>> image = ...
>>> detections = sv.Detections(...)
>>> classification_annotator = sv.ClassificationAnnotator(text_position=sv.Position.CENTER)
>>> classification_annotator = sv.ClassificationAnnotator(
... text_position=sv.Position.CENTER
... )
>>> annotated_frame = classification_annotator.annotate(
... scene=image.copy(),
... detections=detections
@ -1007,7 +1009,9 @@ class ClassificationAnnotator:
self.text_color = color
text = f"{labels[classification_idx]} ({classifications.confidence[classification_idx] * 100:.2f}%)"
percent_confidence = classifications.confidence[classification_idx] * 100
text = f"{labels[classification_idx]} ({percent_confidence:.2f}%)"
text_wh = cv2.getTextSize(
text=text,