insect region mask

app.py

@@ -54,8 +54,6 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
     yellow_background = np.full(image_cv2.shape, (0, 255, 255), dtype=np.uint8)
     
     for detection in detection_results:
-        label = detection.label
-        score = detection.score
         box = detection.box
         mask = detection.mask
         
@@ -63,16 +61,12 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
             mask_uint8 = (mask * 255).astype(np.uint8)
             contours, _ = cv2.findContours(mask_uint8, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
             cv2.drawContours(yellow_background, contours, -1, (0, 0, 0), cv2.FILLED)
-        
-        # Drawing bounding box (border and inside should be yellow)
-        cv2.rectangle(yellow_background, (box.xmin, box.ymin), (box.xmax, box.ymax), (0, 255, 255), cv2.FILLED)  # Fill inside with yellow
-        cv2.rectangle(yellow_background, (box.xmin, box.ymin), (box.xmax, box.ymax), (0, 255, 255), 2)  # Draw border yellow
 
-        if mask is not None:
-            # Overlay the insect on top of the yellow background
-            insect_region = image_cv2[box.ymin:box.ymax, box.xmin:box.xmax]
-            yellow_background[box.ymin:box.ymax, box.xmin:box.xmax] = cv2.bitwise_and(yellow_background[box.ymin:box.ymax, box.xmin:box.xmax], insect_region, mask=mask[box.ymin:box.ymax, box.xmin:box.xmax])
-    
+            # Extract insect region using mask
+            insect_region = cv2.bitwise_and(image_cv2, image_cv2, mask=mask)
+
+            yellow_background[mask > 0] = insect_region[mask > 0]
+
     return cv2.cvtColor(yellow_background, cv2.COLOR_BGR2RGB)
 
 def plot_detections(image: Union[Image.Image, np.ndarray], detections: List[DetectionResult]) -> np.ndarray: