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

fix linters

Browse Source
This commit is contained in:
Daniel Gatis 2024-01-22 22:10:05 -03:00
parent 8347c8b7dc
commit 49d1686f65
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rembg/bg.py
View File

@ -157,7 +157,7 @@ def post_process(mask: np.ndarray) -> np.ndarray:
"""
mask = morphologyEx(mask, MORPH_OPEN, kernel)
mask = GaussianBlur(mask, (5, 5), sigmaX=2, sigmaY=2, borderType=BORDER_DEFAULT)
mask = np.where(mask < 127, 0, 255).astype(np.uint8) # convert again to binary
mask = np.where(mask < 127, 0, 255).astype(np.uint8) # type: ignore
return mask