debug

examples/blobctrl/blobctrl_app.py

@@ -1030,10 +1030,12 @@ def segmentation(img, sel_pix):
     mobile_predictor.set_image(img if isinstance(img, np.ndarray) else np.array(img))
     with torch.no_grad():
         masks, _, _ = mobile_predictor.predict(point_coords=np.array(points), point_labels=np.array(labels), multimask_output=False)
-
+    print("================")
+    print(masks)
+    print("================")
     output_mask = np.ones((masks.shape[1], masks.shape[2], 3))*255
     for i in range(3):
-            output_mask[masks[0] == True, i] = 0.0
+        output_mask[masks[0] == True, i] = 0.0
 
     mask_all = np.ones((masks.shape[1], masks.shape[2], 3))
     color_mask = np.random.random((1, 3)).tolist()[0]
@@ -1056,6 +1058,8 @@ def get_point(img, sel_pix, evt: gr.SelectData):
     print(output_mask.shape)
     print(masked_img)
     print(output_mask)
+    print(np.unique(output_mask))
+
     return masked_img.astype(np.uint8), output_mask