Removed all the alpha matting stuff because bgr will be removed by the caller

hy3dshape/hy3dshape/rembg.py

@@ -38,7 +38,7 @@ from PIL import Image
 from rembg import remove, new_session
 
 
-def _remove_bg_in_process(image_bytes, use_alpha_matting=True):
+def _remove_bg_in_process(image_bytes):
     """Function to run in a separate process to avoid OpenMP issues"""
     try:
         # Create a new session in this process
@@ -47,26 +47,12 @@ def _remove_bg_in_process(image_bytes, use_alpha_matting=True):
         # Convert bytes back to PIL Image
         image = Image.open(io.BytesIO(image_bytes))
         
-        if use_alpha_matting:
-            # High quality mode with alpha matting
-            output = remove(
-                image,
-                session=session,
-                bgcolor=[255, 255, 255, 0],
-                alpha_matting=True,
-                alpha_matting_foreground_threshold=220,
-                alpha_matting_background_threshold=5,
-                alpha_matting_erode_structure_size=5,
-                alpha_matting_base_size=512,
-            )
-        else:
-            # Fallback mode without alpha matting
-            output = remove(
-                image,
-                session=session,
-                bgcolor=[255, 255, 255, 0],
-                alpha_matting=False,
-            )
+        # Standard background removal
+        output = remove(
+            image,
+            session=session,
+            bgcolor=[255, 255, 255, 0],
+        )
         
         # Convert back to bytes
         output_bytes = io.BytesIO()
@@ -114,10 +100,9 @@ class BackgroundRemover():
             if image.mode != 'RGBA':
                 image = image.convert('RGBA')
             
-            print("Starting direct background removal (OpenMP conflicts handled by environment variables)...")
+            print("Starting background removal (OpenMP conflicts handled by environment variables)...")
             
-            # Try alpha matting first (high quality)
-            print("Attempting alpha matting background removal...")
+            # Standard background removal
             try:
                 # Use pre-loaded session if available, otherwise create new one
                 session = self.session if self.session is not None else new_session('u2net')
@@ -125,47 +110,16 @@ class BackgroundRemover():
                     image,
                     session=session,
                     bgcolor=[255, 255, 255, 0],
-                    alpha_matting=True,
-                    alpha_matting_foreground_threshold=220,
-                    alpha_matting_background_threshold=5,
-                    alpha_matting_erode_structure_size=5,
-                    alpha_matting_base_size=512,
                 )
-                print("Alpha matting successful!")
+                print("Background removal successful!")
                 return output
             except Exception as e:
                 error_msg = str(e)
-                print(f"Alpha matting failed: {e}")
+                print(f"Background removal failed: {e}")
                 
                 # Check for specific OpenMP fork errors
                 if "fork()" in error_msg and "OpenMP" in error_msg:
-                    print("DETECTED: OpenMP fork() error in rembg library itself!")
-                    print("This is an internal rembg issue, not our code.")
-                    print("Falling back to standard background removal...")
-                else:
-                    import traceback
-                    traceback.print_exc()
-            
-            # If alpha matting fails, try standard background removal
-            print("Alpha matting failed, falling back to standard background removal")
-            try:
-                # Use pre-loaded session if available, otherwise create new one
-                session = self.session if self.session is not None else new_session('u2net')
-                output = remove(
-                    image,
-                    session=session,
-                    bgcolor=[255, 255, 255, 0],
-                    alpha_matting=False,
-                )
-                print("Standard background removal successful!")
-                return output
-            except Exception as e:
-                error_msg = str(e)
-                print(f"Standard background removal failed: {e}")
-                
-                # Check for specific OpenMP fork errors
-                if "fork()" in error_msg and "OpenMP" in error_msg:
-                    print("DETECTED: OpenMP fork() error in rembg library (standard mode)!")
+                    print("DETECTED: OpenMP fork() error in rembg library!")
                     print("The rembg library itself has internal multiprocessing that conflicts with OpenMP.")
                     print("This is a known issue with the rembg library in daemon environments.")
                 else: