From 5e45468f1b78bbec95815f158563994fbf68fe4f Mon Sep 17 00:00:00 2001 From: chflame163 <130118553+chflame163@users.noreply.github.com> Date: Fri, 26 Jan 2024 22:43:11 +0800 Subject: [PATCH] fix bug --- py/imagefunc.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/py/imagefunc.py b/py/imagefunc.py index 9c9289d..69ab076 100644 --- a/py/imagefunc.py +++ b/py/imagefunc.py @@ -149,7 +149,8 @@ def __rotate_expand(image:Image, angle:float, SSAA:int=0, method:str="lanczos") resize_sampler = Image.NEAREST rotate_sampler = Image.NEAREST if SSAA > 1: - img = tensor.tensor_to_image() + # img = tensor.tensor_to_image() + img = tensor2pil(tensor) img_us_scaled = img.resize((width * SSAA, height * SSAA), resize_sampler) img_rotated = img_us_scaled.rotate(angle, rotate_sampler, expand == "true", fillcolor=(0, 0, 0, 0)) img_down_scaled = img_rotated.resize((img_rotated.width // SSAA, img_rotated.height // SSAA), resize_sampler)