Upload app.py

app.py

@@ -57,21 +57,25 @@ def run_sam(image, predictor, selected_points):
     调用 SAM 模型进行分割。
     """
     print(image.shape)
-    print(np.unique(image))
+    print(image[0,0,:])
+    print(np.unique(image)) 
     predictor.set_image(image)
     if len(selected_points) == 0:
         return [], None
-    input_points = [p for p in selected_points]
-    input_labels = [1 for _ in range(len(selected_points))]
+    # input_points = [p for p in selected_points]
+    # input_labels = [1 for _ in range(len(selected_points))]
+    input_points = np.array([[210, 300]])
+    input_labels = np.array([1])
     print(input_points)
     print(input_labels)
     masks, _, _ = predictor.predict(
-        point_coords=np.array(input_points),
-        point_labels=np.array(input_labels),
-        multimask_output=False,  # 单对象输出
+        point_coords=input_points,
+        point_labels=input_labels,
+        multimask_output=True,  # 单对象输出
     )
     print(masks.shape, np.unique(masks))
-    visible_mask = 255 * np.squeeze(masks).astype(np.uint8)
+    best_mask = masks[0]
+    visible_mask = 255 * best_mask.astype(np.uint8)
     return visible_mask
 
 def apply_mask_overlay(image, mask):