Update ImageNode.py

shadowcz007 · Jan 25, 2024 · 3d68e48 · 3d68e48
1 parent 4351fa6
commit 3d68e48
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py
@@ -2120,14 +2120,15 @@ def run(self,width,height,scale_option,image=None,average_color=['on'],fill_colo
                     average_images.append(a_im)
                     hexs.append(hex)
 
-            for mas in mask:
-                for ma in mas:
-                    ma=tensor2pil(ma)
-                    ma=ma.convert('RGB')
-                    ma=resize_image(ma,scale_option,w,h,fill_color)
-                    ma=ma.convert('L')
-                    ma=pil2tensor(ma)
-                    masks.append(ma)
+            if mask!=None:
+                for mas in mask:
+                    for ma in mas:
+                        ma=tensor2pil(ma)
+                        ma=ma.convert('RGB')
+                        ma=resize_image(ma,scale_option,w,h,fill_color)
+                        ma=ma.convert('L')
+                        ma=pil2tensor(ma)
+                        masks.append(ma)
 
         return (imgs,average_images,hexs,masks,)