From b7915ac877da927d4effc4a64235e03ade046aa9 Mon Sep 17 00:00:00 2001 From: Patrick Morgan Date: Sun, 15 Dec 2024 22:01:31 -0800 Subject: [PATCH] WIP --- nodes/nodes.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nodes/nodes.py b/nodes/nodes.py index d9cf79a..5fcd1d8 100644 --- a/nodes/nodes.py +++ b/nodes/nodes.py @@ -410,9 +410,9 @@ class MaskContourProcessor: combined_mask = np.clip(mask_np + output_mask, 0, 1) - # Apply Gaussian blur - blurred_image = Image.fromarray(combined_mask).filter(ImageFilter.GaussianBlur(10)) - blurred_mask = np.array(blurred_image) + # Convert to 'L' mode and apply Gaussian blur + blurred_image = Image.fromarray((combined_mask * 255).astype(np.uint8), mode='L').filter(ImageFilter.GaussianBlur(10)) + blurred_mask = np.array(blurred_image) / 255.0 return (torch.from_numpy(blurred_mask),)