Fix batching images

This commit is contained in:
Alexis Rolland
2024-11-21 21:26:40 +08:00
parent bc59424ab0
commit b3db70e371
+2 -3
View File
@@ -39,13 +39,12 @@ class DropShadowV2:
background_image=None, layer_mask=None
):
# If background image is empty, create transparent background
# image for each layer image
# If background image is empty, create transparent background image for each layer image
if background_image == None:
background_image = []
for l in layer_image:
m = tensor2pil(l)
background_image.append(pil2tensor(Image.new('RGBA', (m.width, m.height), (0, 0, 0, 0))))
background_image.append(pil2tensor(Image.new('RGBA', (m.width, m.height), (0, 0, 0, 0))))
b_images = []
l_images = []