annotate_with_transparency

app.py

@@ -57,17 +57,17 @@ def annotate_with_transparency(image: Union[Image.Image, np.ndarray], detection_
         mask = detection.mask
         color = np.random.randint(0, 256, size=3).tolist()
 
-        # 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)
+            
+            # Create an overlay with transparent background
+            overlay = image_cv2.copy()
             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)
+            # Blend the overlay with the original image
+            alpha = 0.5  # Adjust the alpha value to control transparency
+            image_cv2 = cv2.addWeighted(overlay, alpha, image_cv2, 1 - alpha, 0)
 
         cv2.putText(image_cv2, f'{label}: {score:.2f}', (box.xmin, box.ymin - 10),
                     cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)