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

fix lint

Browse Source
This commit is contained in:
Daniel Gatis 2023-07-13 03:16:12 -03:00
parent f0019d723b
commit d54b62a8b1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rembg/sessions/u2net_cloth_seg.py
View File

@ -82,13 +82,13 @@ class Unet2ClothSession(BaseSession):
mask1.putpalette(palette1)
mask1 = mask1.convert("RGB").convert("L")
masks.append(mask1)
def lower_cloth():
mask2 = mask.copy()
mask2.putpalette(palette2)
mask2 = mask2.convert("RGB").convert("L")
masks.append(mask2)
def full_cloth():
mask3 = mask.copy()
mask3.putpalette(palette3)