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

View File

@ -1455,7 +1455,7 @@ class ImageAlphaComposite:
mask1[0, :, :, 2] = mask mask1[0, :, :, 2] = mask
print(image1.size(), mask1.size()) print(image1.size(), mask1.size())
image = image1 * mask1 + image2 * (torch.ones_like(mask) - mask) image = image1 * mask1 + image2 * (torch.ones_like(mask1) - mask1)
return (image,) return (image,)
class ImageScaleBy: class ImageScaleBy: