From ff0f2e3984136cb27e76bad033c04006fe6693e1 Mon Sep 17 00:00:00 2001 From: Favorfitai Date: Tue, 13 Aug 2024 09:34:57 +0900 Subject: [PATCH] Update: lanzcos resize --- utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils.py b/utils.py index 4c75478..1202afe 100644 --- a/utils.py +++ b/utils.py @@ -1,6 +1,7 @@ import torch from torchvision import transforms from comfy import model_management +from PIL import Image def simple_resize(image_tensor:torch.Tensor, height, width): """ @@ -117,7 +118,7 @@ def dynamic_resize(image_tensor:torch.Tensor, max_pixels:int=1024*1024, min_pixe else: return image_tensor - transform_resize = transforms.Resize((new_height, new_width)) + transform_resize = transforms.Resize((new_height, new_width), interpolation=Image.LANCZOS) resized_ts = transform_resize(img_ts) # 원래 데이터 형태로 복구