From f16d576f6f0e83fc2fafd2d1f29b2edeb00d3197 Mon Sep 17 00:00:00 2001
From: Mel Massadian <mel@melmassadian.com>
Date: Tue, 2 Apr 2024 05:33:16 +0200
Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20image=20to=20mask?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes/image_processing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes/image_processing.py b/nodes/image_processing.py
index 6b71a99..608f42f 100644
--- a/nodes/image_processing.py
+++ b/nodes/image_processing.py
@@ -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")