update default sam_prompt structure to include point data

Signed-off-by: ふぁ <[email protected]>
danielgatis · Nov 29, 2024 · 9e6c461 · 9e6c461
1 parent 00ad95d
commit 9e6c461
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/rembg/sessions/sam.py b/rembg/sessions/sam.py
@@ -143,7 +143,16 @@ def predict(
         Returns:
             List[PILImage]: A list of masks generated by the decoder.
         """
-        prompt = kwargs.get("sam_prompt", "{}")
+        prompt = kwargs.get(
+            "sam_prompt",
+            [
+                {
+                    "type": "point",
+                    "label": 1,
+                    "data": [int(img.width / 2), int(img.height / 2)],
+                }
+            ],
+        )
         schema = {
             "type": "array",
             "items": {