AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-16 17:15:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix linters

Browse Source
This commit is contained in:
Daniel Gatis 2024-05-23 17:19:43 -03:00
parent f13083b55d
commit 747ef5e793
9 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rembg/sessions/base.py
View File

@ -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)

2
rembg/sessions/dis_anime.py
View File

@ -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]

2
rembg/sessions/dis_general_use.py
View File

@ -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]

2
rembg/sessions/silueta.py
View File

@ -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]

2
rembg/sessions/u2net.py
View File

@ -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]

2
rembg/sessions/u2net_cloth_seg.py
View File

@ -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 = []

2
rembg/sessions/u2net_custom.py
View File

@ -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]

2
rembg/sessions/u2net_human_seg.py
View File

@ -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]

2
rembg/sessions/u2netp.py
View File

@ -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]