Update app.py

app.py

@@ -46,7 +46,7 @@ class DetectionResult:
             )
         )
 
-def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[DetectionResult]) -> np.ndarray:
+def annotate_with_transparency(image: Union[Image.Image, np.ndarray], detection_results: List[DetectionResult]) -> np.ndarray:
     image_cv2 = np.array(image) if isinstance(image, Image.Image) else image
     image_cv2 = cv2.cvtColor(image_cv2, cv2.COLOR_RGB2BGR)
 
@@ -57,19 +57,25 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
         mask = detection.mask
         color = np.random.randint(0, 256, size=3).tolist()
 
-        cv2.rectangle(image_cv2, (box.xmin, box.ymin), (box.xmax, box.ymax), color, 2)
-        cv2.putText(image_cv2, f'{label}: {score:.2f}', (box.xmin, box.ymin - 10),
-                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
+        # Overlay with transparent background
+        overlay = np.zeros_like(image_cv2, dtype=np.uint8)
 
         if mask is not None:
             mask_uint8 = (mask * 255).astype(np.uint8)
             contours, _ = cv2.findContours(mask_uint8, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-            cv2.drawContours(image_cv2, contours, -1, color, 2)
+            cv2.drawContours(overlay, contours, -1, color, -1)  # Draw filled contours on the overlay
+
+        # Blend the overlay with the original image
+        alpha = 0.5  # Alpha value to control transparency
+        image_cv2 = cv2.addWeighted(image_cv2, 1, overlay, alpha, 0)
+
+        cv2.putText(image_cv2, f'{label}: {score:.2f}', (box.xmin, box.ymin - 10),
+                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
 
     return cv2.cvtColor(image_cv2, cv2.COLOR_BGR2RGB)
 
 def plot_detections(image: Union[Image.Image, np.ndarray], detections: List[DetectionResult]) -> np.ndarray:
-    annotated_image = annotate(image, detections)
+    annotated_image = annotate_with_transparency(image, detections)
     return annotated_image
 
 def load_image(image: Union[str, Image.Image]) -> Image.Image: