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

fix issue 116

Browse Source
This commit is contained in:
Daniel Gatis 2021-07-06 00:56:31 -03:00
parent 4e64eccb44
commit ae46588c7e
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -11,7 +11,7 @@ with open("requirements.txt") as f:
setup(
name="rembg",
version="1.0.26",
version="1.0.27",
description="Remove image background",
long_description=long_description,
long_description_content_type="text/markdown",

19
src/rembg/bg.py
View File

@ -92,14 +92,17 @@ def remove(
mask = detect.predict(model, np.array(img)).convert("L")
if alpha_matting:
cutout = alpha_matting_cutout(
img,
mask,
alpha_matting_foreground_threshold,
alpha_matting_background_threshold,
alpha_matting_erode_structure_size,
alpha_matting_base_size,
)
try:
cutout = alpha_matting_cutout(
img,
mask,
alpha_matting_foreground_threshold,
alpha_matting_background_threshold,
alpha_matting_erode_structure_size,
alpha_matting_base_size,
)
except:
cutout = naive_cutout(img, mask)
else:
cutout = naive_cutout(img, mask)