From 747ef5e7938bb7b29581f4b7acc0f420e8734557 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Thu, 23 May 2024 17:19:43 -0300 Subject: [PATCH] fix linters --- rembg/sessions/base.py | 2 +- rembg/sessions/dis_anime.py | 2 +- rembg/sessions/dis_general_use.py | 2 +- rembg/sessions/silueta.py | 2 +- rembg/sessions/u2net.py | 2 +- rembg/sessions/u2net_cloth_seg.py | 2 +- rembg/sessions/u2net_custom.py | 2 +- rembg/sessions/u2net_human_seg.py | 2 +- rembg/sessions/u2netp.py | 2 +- 9 files changed, 9 insertions(+), 9 deletions(-) diff --git a/rembg/sessions/base.py b/rembg/sessions/base.py index f942840..ec9c037 100644 --- a/rembg/sessions/base.py +++ b/rembg/sessions/base.py @@ -46,7 +46,7 @@ class BaseSession: *args, **kwargs ) -> Dict[str, np.ndarray]: - im = img.convert("RGB").resize(size, Image.LANCZOS) + im = img.convert("RGB").resize(size, Image.Resampling.LANCZOS) im_ary = np.array(im) im_ary = im_ary / np.max(im_ary) diff --git a/rembg/sessions/dis_anime.py b/rembg/sessions/dis_anime.py index 01e9a2a..33d09a7 100644 --- a/rembg/sessions/dis_anime.py +++ b/rembg/sessions/dis_anime.py @@ -40,7 +40,7 @@ class DisSession(BaseSession): pred = np.squeeze(pred) mask = Image.fromarray((pred * 255).astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) return [mask] diff --git a/rembg/sessions/dis_general_use.py b/rembg/sessions/dis_general_use.py index fb1290d..f19b0ed 100644 --- a/rembg/sessions/dis_general_use.py +++ b/rembg/sessions/dis_general_use.py @@ -36,7 +36,7 @@ class DisSession(BaseSession): pred = np.squeeze(pred) mask = Image.fromarray((pred * 255).astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) return [mask] diff --git a/rembg/sessions/silueta.py b/rembg/sessions/silueta.py index 7851387..574a33c 100644 --- a/rembg/sessions/silueta.py +++ b/rembg/sessions/silueta.py @@ -42,7 +42,7 @@ class SiluetaSession(BaseSession): pred = np.squeeze(pred) mask = Image.fromarray((pred * 255).astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) return [mask] diff --git a/rembg/sessions/u2net.py b/rembg/sessions/u2net.py index 1c9359e..9cf8cf6 100644 --- a/rembg/sessions/u2net.py +++ b/rembg/sessions/u2net.py @@ -42,7 +42,7 @@ class U2netSession(BaseSession): pred = np.squeeze(pred) mask = Image.fromarray((pred * 255).astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) return [mask] diff --git a/rembg/sessions/u2net_cloth_seg.py b/rembg/sessions/u2net_cloth_seg.py index b3587d1..aaac286 100644 --- a/rembg/sessions/u2net_cloth_seg.py +++ b/rembg/sessions/u2net_cloth_seg.py @@ -87,7 +87,7 @@ class Unet2ClothSession(BaseSession): pred = np.squeeze(pred, 0) mask = Image.fromarray(pred.astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) masks = [] diff --git a/rembg/sessions/u2net_custom.py b/rembg/sessions/u2net_custom.py index b984208..a513917 100644 --- a/rembg/sessions/u2net_custom.py +++ b/rembg/sessions/u2net_custom.py @@ -68,7 +68,7 @@ class U2netCustomSession(BaseSession): pred = np.squeeze(pred) mask = Image.fromarray((pred * 255).astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) return [mask] diff --git a/rembg/sessions/u2net_human_seg.py b/rembg/sessions/u2net_human_seg.py index 8c80db1..170e5ab 100644 --- a/rembg/sessions/u2net_human_seg.py +++ b/rembg/sessions/u2net_human_seg.py @@ -42,7 +42,7 @@ class U2netHumanSegSession(BaseSession): pred = np.squeeze(pred) mask = Image.fromarray((pred * 255).astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) return [mask] diff --git a/rembg/sessions/u2netp.py b/rembg/sessions/u2netp.py index 94b0f16..2988afd 100644 --- a/rembg/sessions/u2netp.py +++ b/rembg/sessions/u2netp.py @@ -38,7 +38,7 @@ class U2netpSession(BaseSession): pred = np.squeeze(pred) mask = Image.fromarray((pred * 255).astype("uint8"), mode="L") - mask = mask.resize(img.size, Image.LANCZOS) + mask = mask.resize(img.size, Image.Resampling.LANCZOS) return [mask]