updating text when detecting new class

yassine-rd · Nov 29, 2022 · e4652eb · e4652eb
1 parent 6c1bbcb
commit e4652eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/utils.py b/core/utils.py
@@ -177,7 +177,7 @@ def draw_bbox(image, bboxes, info = False, counted_classes = None, show_label=Tr
                 height_ratio = int(image_h / 25)
                 offset = 15
                 for key, value in counted_classes.items():
-                    cv2.putText(image, "{}s detected: {}".format(key, value), (5, offset),
+                    cv2.putText(image, "{} detected: {}".format(key, value), (5, offset),
                             cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (0, 255, 0), 2)
                     offset += height_ratio
     return image