fix SamAutoMaskSEGS result

lldacing · Nov 11, 2024 · 94885d3 · 94885d3
1 parent 79a5fef
commit 94885d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easyapi/SamNode.py b/easyapi/SamNode.py
@@ -230,7 +230,7 @@ def INPUT_TYPES(self):
 
     def generate(self, sam_model, image, output_mode):
         masks = super().generate(sam_model, image, output_mode=output_mode)
-        masksRle = json.JSONEncoder().encode(masks)
+        masksRle = json.JSONEncoder().encode(masks[0])
         return {"ui": {"segsRle": (masksRle,)}, "result": (masksRle,)}