experimenting

app.py

@@ -51,20 +51,12 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
     image_cv2 = cv2.cvtColor(image_cv2, cv2.COLOR_RGB2BGR)
 
     for detection in detection_results:
-        label = detection.label
-        score = detection.score
-        box = detection.box
         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)
-
         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)
+            # Drawing only the mask contours without any bounding box or label
+            cv2.drawContours(image_cv2, contours, -1, (0, 0, 0), 2)  # Black color for contours
 
     return cv2.cvtColor(image_cv2, cv2.COLOR_BGR2RGB)
 
@@ -172,8 +164,7 @@ def run_length_encoding(mask):
                 rle.append(count)
             count = 1
             last_val = pixel
-    if count > 0:
-        rle.append(count)
+    if count > 0, rle.append(count)
     return rle
 
 def detections_to_json(detections):