From 4eb2a77475a69b5ce6b35a4d875f3b75ce95a977 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Sat, 8 Apr 2023 06:47:05 +0900
Subject: [PATCH] length function patch

---
 comfyui-impact-pack.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/comfyui-impact-pack.py b/comfyui-impact-pack.py
index 5daf351..fa60991 100644
--- a/comfyui-impact-pack.py
+++ b/comfyui-impact-pack.py
@@ -101,7 +101,7 @@ def create_segmasks(results):
 
 
 def combine_masks(masks):
-    if masks.__len__ == 0:
+    if len(masks) == 0:
         return None
     else:
         initial_cv2_mask = np.array(masks[0][1])
@@ -116,7 +116,7 @@ def combine_masks(masks):
 
 
 def combine_masks2(masks):
-    if masks.__len__ == 0:
+    if len(masks) == 0:
         return None
     else:
         initial_cv2_mask = np.array(masks[0]).astype(np.uint8)
@@ -632,7 +632,10 @@ class SAMDetectorCombined:
             if not selected:
                 total_masks.append(max_mask)
 
-        mask = combine_masks2(total_masks).float()
+        mask = combine_masks2(total_masks)
+        if mask is not None:
+            mask = mask.float()
+
         return (mask, )