From f25f6b012488b24ced518822fae9fd7200e724be Mon Sep 17 00:00:00 2001 From: yuyuyzl Date: Sat, 16 Aug 2025 05:56:01 +0800 Subject: [PATCH] feat: revert pixel 0,0,0,0 preprocess --- main.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/main.py b/main.py index 8376d9f..c22a380 100644 --- a/main.py +++ b/main.py @@ -534,6 +534,11 @@ def main(): img = Image.open(f"data/images/{args.character}.png") img = img.convert('RGBA') ow, oh = img.size + for i, px in enumerate(img.getdata()): + if px[3] <= 0: + y = i // ow + x = i % ow + img.putpixel((x, y), (0, 0, 0, 0)) if ow != 512 or oh != 512: img = resize_to_512_center(img) if args.alpha_clean: