add alpha matting

This commit is contained in:
Daniel Gatis 2020-10-09 15:36:25 -03:00
parent 135550812c
commit 01f1769493
3 changed files with 7 additions and 7 deletions

View File

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

View File

@ -70,9 +70,9 @@ def remove(
data,
model_name="u2net",
alpha_matting=False,
alpha_matting_foreground_threshold=235,
alpha_matting_background_threshold=15,
alpha_matting_erode_structure_size=15,
alpha_matting_foreground_threshold=240,
alpha_matting_background_threshold=10,
alpha_matting_erode_structure_size=10,
):
model = model_u2net

View File

@ -32,7 +32,7 @@ def main():
ap.add_argument(
"-af",
"--alpha-matting-foreground-threshold",
default=235,
default=240,
type=int,
help="The trimap foreground threshold.",
)
@ -40,7 +40,7 @@ def main():
ap.add_argument(
"-ab",
"--alpha-matting-background-threshold",
default=15,
default=10,
type=int,
help="The trimap background threshold.",
)
@ -48,7 +48,7 @@ def main():
ap.add_argument(
"-ae",
"--alpha-matting-erode-size",
default=15,
default=10,
type=int,
help="Size of element used for the erosion.",
)