refactor label scores

Signed-off-by: Ashwin Vaidya <[email protected]>

Author: ashwinvaidya17

src/python/model_api/visualizer/primitive/label.py

@@ -19,6 +19,7 @@ class Label(Primitive):
 
     Args:
         label (str): Text of the label.
+        score (float | None): Score of the label. This is optional.
         fg_color (str | tuple[int, int, int]): Foreground color of the label.
         bg_color (str | tuple[int, int, int]): Background color of the label.
         font_path (str | None | BytesIO): Path to the font file.
@@ -40,13 +41,14 @@ class Label(Primitive):
 
     def __init__(
         self,
-        label: Union[str, float],
+        label: str,
+        score: Union[float, None] = None,
         fg_color: Union[str, tuple[int, int, int]] = "black",
         bg_color: Union[str, tuple[int, int, int]] = "yellow",
         font_path: Union[str, BytesIO, None] = None,
         size: int = 16,
     ) -> None:
-        self.label = str(label)
+        self.label = f"{label} ({score:.2f})" if score is not None else label
         self.fg_color = fg_color
         self.bg_color = bg_color
         self.font = ImageFont.load_default(size=size) if font_path is None else ImageFont.truetype(font_path, size)

src/python/model_api/visualizer/scene/anomaly.py

@@ -42,10 +42,8 @@ def _get_bounding_boxes(self, result: AnomalyResult) -> list[BoundingBox]:
 
     def _get_labels(self, result: AnomalyResult) -> list[Label]:
         labels = []
-        if result.pred_label is not None:
-            labels.append(Label(result.pred_label))
-        if result.pred_score is not None:
-            labels.append(Label(result.pred_score))
+        if result.pred_label is not None and result.pred_score is not None:
+            labels.append(Label(label=result.pred_label, score=result.pred_score))
         return labels
 
     def _get_polygons(self, result: AnomalyResult) -> list[Polygon]: