diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py index 7f47029a..13eab4a0 100644 --- 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,)