Merge pull request #71 from husejinbg/main

replaced deprecated BoxAnnotator with BoundingBoxAnnotator and LabelA…
hkchengrex · Mar 19, 2024 · 58cff8c · 58cff8c
2 parents 04a7eff + 4b35c7a
commit 58cff8c
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/deva/inference/result_utils.py b/deva/inference/result_utils.py
@@ -260,10 +260,13 @@ def save_result(queue: Queue):
                         detections = sv.Detections(xyxy,
                                                    confidence=np.array(all_scores),
                                                    class_id=np.array(all_cat_ids))
-                        annotator = sv.BoxAnnotator()
-                        blend = annotator.annotate(scene=blend,
-                                                   detections=detections,
-                                                   labels=labels)
+                        box_annotator = sv.BoundingBoxAnnotator()
+                        label_annotator = sv.LabelAnnotator()
+                        blend = box_annotator.annotate(scene=blend,
+                                                   detections=detections)
+                        blend = label_annotator.annotate(scene=blend,
+                                                        detections=detections,
+                                                        labels=labels)
 
                 if saver.dataset != 'gradio':
                     # find a place to save the visualization