fix SamAutoMaskSEGS result

2024-11-11 13:32:54 +08:00
parent 79a5fefc5d
commit 94885d3ac5
1 changed files with 1 additions and 1 deletions
@@ -230,7 +230,7 @@ class SamAutoMaskSEGS(SamAutoMaskSEGSAdvanced):

    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,)}