From 94885d3ac5f043c98992b314602f28fcccdb9425 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E9=9B=AA=E5=B3=B0?= Date: Mon, 11 Nov 2024 13:32:54 +0800 Subject: [PATCH] fix SamAutoMaskSEGS result --- easyapi/SamNode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/easyapi/SamNode.py b/easyapi/SamNode.py index 8546c62..2126df4 100644 --- a/easyapi/SamNode.py +++ b/easyapi/SamNode.py @@ -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,)}