Update ImageNode.py

2024-01-25 20:10:29 +08:00
parent 3d68e48219
commit 3a3da49c69
1 changed files with 3 additions and 1 deletions
@@ -2120,7 +2120,7 @@ class ResizeImage:
                    average_images.append(a_im)
                    hexs.append(hex)

-            if mask!=None:
+            try:
                for mas in mask:
                    for ma in mas:
                        ma=tensor2pil(ma)
@@ -2129,6 +2129,8 @@ class ResizeImage:
                        ma=ma.convert('L')
                        ma=pil2tensor(ma)
                        masks.append(ma)
+            except:
+                print('')
        
        return (imgs,average_images,hexs,masks,)