Browse Source

fix issue 116

Daniel Gatis 4 years ago
parent
commit
ae46588c7e
2 changed files with 12 additions and 9 deletions
  1. 1 1
      setup.py
  2. 11 8
      src/rembg/bg.py

+ 1 - 1
setup.py

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

+ 11 - 8
src/rembg/bg.py

@@ -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(
-            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:
     else:
         cutout = naive_cutout(img, mask)
         cutout = naive_cutout(img, mask)