From b3db70e371770ebcca6d22961962c0e8a3b3a6e5 Mon Sep 17 00:00:00 2001 From: Alexis Rolland Date: Thu, 21 Nov 2024 21:26:40 +0800 Subject: [PATCH] Fix batching images --- py/drop_shadow_v2.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/py/drop_shadow_v2.py b/py/drop_shadow_v2.py index c4d9508..4ecff09 100644 --- a/py/drop_shadow_v2.py +++ b/py/drop_shadow_v2.py @@ -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 = []