This commit is contained in:
Saquib Alam 2023-08-08 21:12:09 +05:30 committed by GitHub
parent 17b2999195
commit 3fe543dedc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1448,14 +1448,15 @@ class ImageAlphaComposite:
RETURN_TYPES = ("IMAGE",)
FUNCTION = "alpha_composite"
def alpha_composite(self, image1, mask, image2):
# image1 = torch.nn.functional.interpolate(mask[None, None], size=(source.shape[2], source.shape[3]), mode="bilinear")
mask1 = torch.ones_like(image1)
mask = torch.ones_like(mask) - mask
mask1[0, :, :, 0] = mask
mask1[0, :, :, 1] = mask
mask1[0, :, :, 2] = mask
print(image1.size(), mask1.size())
image = image1 * mask1 + image2 * (torch.ones_like(mask1) - mask1)
image = image1 * (torch.ones_like(mask1) - mask1)
return (image,)
class ImageScaleBy: