From e9f9a50d6e909033c3c3ecc8c4135cfa5c4dbe25 Mon Sep 17 00:00:00 2001 From: Saquib Alam Date: Tue, 8 Aug 2023 22:39:13 +0530 Subject: [PATCH] smooth image from LoadImage --- nodes.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index 909a3d76f..fe736b8bc 100644 --- a/nodes.py +++ b/nodes.py @@ -1332,7 +1332,7 @@ class LoadImage: CATEGORY = "image" - RETURN_TYPES = ("IMAGE", "MASK") + RETURN_TYPES = ("IMAGE", "MASK", "IMAGE") FUNCTION = "load_image" def load_image(self, image): image_path = folder_paths.get_annotated_filepath(image) @@ -1341,12 +1341,21 @@ class LoadImage: image = i.convert("RGB") image = np.array(image).astype(np.float32) / 255.0 image = torch.from_numpy(image)[None,] + + smooth_image = image.copy() if 'A' in i.getbands(): mask = np.array(i.getchannel('A')).astype(np.float32) / 255.0 - mask = 1. - torch.from_numpy(mask) + mask = torch.from_numpy(mask) + + mask1 = torch.ones_like(image) + mask1[0, :, :, 0] = mask + mask1[0, :, :, 1] = mask + mask1[0, :, :, 2] = mask + smooth_image = smooth_image * mask + mask = 1.0 - mask else: mask = torch.zeros((64,64), dtype=torch.float32, device="cpu") - return (image, mask) + return (image, mask, smooth_image) @classmethod def IS_CHANGED(s, image):