modified segmentation mask display

inference.py

@@ -70,7 +70,7 @@ def draw_boxes(
 def draw_masks(
     frame: np.ndarray,
     masks: np.ndarray,
-    alpha: float = 0.45,
+    alpha: float = 0.65,
     labels: Optional[Sequence[str]] = None,
 ) -> np.ndarray:
     output = frame.copy()
@@ -93,6 +93,11 @@ def draw_masks(
         color = _color_for_label(label)
         overlay[mask_bool] = color
         output = cv2.addWeighted(output, 1.0, overlay, alpha, 0)
+        contours, _ = cv2.findContours(
+            mask_bool.astype(np.uint8), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
+        )
+        if contours:
+            cv2.drawContours(output, contours, -1, color, thickness=2)
         if label:
             coords = np.column_stack(np.where(mask_bool))
             if coords.size:
@@ -198,6 +203,8 @@ def infer_segmentation_frame(
     with lock:
         result = segmenter.predict(frame, text_prompts=text_queries)
     labels = text_queries or []
+    if len(labels) == 1:
+        labels = [labels[0] for _ in range(len(result.masks or []))]
     return draw_masks(frame, result.masks, labels=labels), result