diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py index 64a4ac1..e03960b 100644 --- a/src/aiunn/finetune.py +++ b/src/aiunn/finetune.py @@ -84,8 +84,8 @@ class UpscaleDataset(Dataset): # Resize the images to reduce VRAM usage. # Using Image.ANTIALIAS which is equivalent to LANCZOS in current Pillow versions. - low_res = low_res.resize((384, 384), Image.ANTIALIAS) - high_res = high_res.resize((768, 768), Image.ANTIALIAS) + low_res = low_res.resize((384, 384), Image.LANCZOS) + high_res = high_res.resize((768, 768), Image.LANCZOS) # If a transform is provided (e.g. conversion to Tensor), apply it. if self.transform: