Ver código fonte

add alpha matting

Daniel Gatis 4 anos atrás
pai
commit
01f1769493
3 arquivos alterados com 7 adições e 7 exclusões
  1. 1 1
      setup.py
  2. 3 3
      src/rembg/bg.py
  3. 3 3
      src/rembg/cmd/cli.py

+ 1 - 1
setup.py

@@ -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",

+ 3 - 3
src/rembg/bg.py

@@ -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
 

+ 3 - 3
src/rembg/cmd/cli.py

@@ -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.",
     )