Przeglądaj źródła

Merge branch 'danielgatis:main' into rembg-updates-4-ben

BlackNoizE404 3 miesięcy temu
rodzic
commit
e25e3e5353
2 zmienionych plików z 1 dodań i 2 usunięć
  1. 0 1
      rembg/sessions/ben_custom.py
  2. 1 1
      rembg/sessions/dis_custom.py

+ 0 - 1
rembg/sessions/ben_custom.py

@@ -60,7 +60,6 @@ class BenCustomSession(BaseSession):
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
         mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
-
         return [mask]
 
     @classmethod

+ 1 - 1
rembg/sessions/dis_custom.py

@@ -8,6 +8,7 @@ from PIL.Image import Image as PILImage
 
 from .base import BaseSession
 
+
 class DisCustomSession(BaseSession):
     """This is a class representing a custom session for the Dis model."""
 
@@ -27,7 +28,6 @@ class DisCustomSession(BaseSession):
 
         super().__init__(model_name, sess_opts, *args, **kwargs)
 
-
     def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         """
         Predicts the mask image for the input image.