debug

examples/blobctrl/blobctrl_app.py

@@ -1025,7 +1025,7 @@ def segmentation(img, sel_pix):
     for p, l in sel_pix:
         points.append(p)
         labels.append(l)
-    mobile_predictor.set_image(img if isinstance(img, np.ndarray) else np.array(img))
+    mobile_predictor.set_image(img.astype(np.uint8) if isinstance(img, np.ndarray) else np.array(img).astype(np.uint8))
     with torch.no_grad():
         masks, _, _ = mobile_predictor.predict(point_coords=np.array(points), point_labels=np.array(labels), multimask_output=False)
     print("=======img=========")
@@ -1715,7 +1715,7 @@ with block as demo:
     ## select point
     input_image.select(
         get_point,
-        [original_image, selected_points],
+        [input_image, selected_points],
         [input_image, original_mask],
     )