Upload app.py

app.py

@@ -48,7 +48,7 @@ def reset_image(predictor, img):
       - 设置 predictor 的输入图像，
       - 返回原图
     """
-    # predictor.set_image(img)
+    predictor.set_image(img)
     original_img = img.copy()
     # 返回predictor，visible occlusion mask初始化, 原始图像
     return predictor, img, img, original_img
@@ -63,14 +63,14 @@ def run_sam(image, predictor, selected_points):
     """
     调用 SAM 模型进行分割。
     """
-    predictor.set_image(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 = np.array([[210, 300]])
-    input_labels = np.array([1])
-    masks, scores, _ = predictor.predict(
+    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])
+    masks, _, _ = predictor.predict(
         point_coords=input_points,
         point_labels=input_labels,
         multimask_output=False,  # 单对象输出