fix mask preview feature

2024-06-08 09:38:26 -04:00
parent 9ebb5732a7
commit 7c22d9743e
2 changed files with 6 additions and 3 deletions
@@ -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)

@@ -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