AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-15 21:45:52 +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( setup(
name="rembg", name="rembg",
version="1.0.26", version="1.0.27",
description="Remove image background", description="Remove image background",
long_description=long_description, long_description=long_description,
long_description_content_type="text/markdown", 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") mask = detect.predict(model, np.array(img)).convert("L")
if alpha_matting: if alpha_matting:
cutout = alpha_matting_cutout( try:
img, cutout = alpha_matting_cutout(
mask, img,
alpha_matting_foreground_threshold, mask,
alpha_matting_background_threshold, alpha_matting_foreground_threshold,
alpha_matting_erode_structure_size, alpha_matting_background_threshold,
alpha_matting_base_size, alpha_matting_erode_structure_size,
) alpha_matting_base_size,
)
except:
cutout = naive_cutout(img, mask)
else: else:
cutout = naive_cutout(img, mask) cutout = naive_cutout(img, mask)