From 7c22d9743eb47d0bdfbbd3f2adc72b3a0bc1688d Mon Sep 17 00:00:00 2001 From: Terry Jia Date: Sat, 8 Jun 2024 09:38:26 -0400 Subject: [PATCH] fix mask preview feature --- layer_divider_node.py | 2 +- ldivider/ld_segment.py | 7 +++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/layer_divider_node.py b/layer_divider_node.py index ab8b920..d27ede9 100644 --- a/layer_divider_node.py +++ b/layer_divider_node.py @@ -283,7 +283,7 @@ class LayerDividerSegmentMask: masked_image = show_anns(input_image, masks, output_dir) - #masked_image = to_comfy_imgs(masked_image) + masked_image = to_comfy_img(np.array(masked_image)) input_image.putalpha(255) diff --git a/ldivider/ld_segment.py b/ldivider/ld_segment.py index 3f600a5..cf7760b 100644 --- a/ldivider/ld_segment.py +++ b/ldivider/ld_segment.py @@ -61,8 +61,11 @@ def show_anns(image, masks, output_dir): mask_list.append(img) base_mask = image - #for mask in mask_list: - # base_mask = Image.alpha_composite(base_mask, Image.fromarray(mask)) + for mask in mask_list: + img1 = base_mask.convert('RGBA') + img2 = Image.fromarray(mask) + + base_mask = Image.alpha_composite(img1, img2) return base_mask