fix: 🐛 image to mask

2024-04-02 05:33:16 +02:00
parent e56508c207
commit f16d576f6f
1 changed files with 1 additions and 1 deletions
@@ -412,7 +412,7 @@ class MTB_MaskToImage:
    FUNCTION = "render_mask"

    def render_mask(self, mask, color, background):
-        masks = tensor2np(mask)
+        masks = tensor2np(mask)[0]
        images = []
        for m in masks:
            _mask = Image.fromarray(m).convert("L")