Kaynağa Gözat

Merge pull request #362 from samiede/main

Update resampling method to allow use of current Pillow version
Daniel Gatis 2 yıl önce
ebeveyn
işleme
d62227d586
2 değiştirilmiş dosya ile 2 ekleme ve 2 silme
  1. 1 1
      rembg/session_base.py
  2. 1 1
      rembg/session_simple.py

+ 1 - 1
rembg/session_base.py

@@ -18,7 +18,7 @@ class BaseSession:
         std: Tuple[float, float, float],
         size: Tuple[int, int],
     ) -> Dict[str, np.ndarray]:
-        im = img.convert("RGB").resize(size, Image.Resampling.LANCZOS)
+        im = img.convert("RGB").resize(size, Image.LANCZOS)
 
         im_ary = np.array(im)
         im_ary = im_ary / np.max(im_ary)

+ 1 - 1
rembg/session_simple.py

@@ -25,6 +25,6 @@ class SimpleSession(BaseSession):
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
+        mask = mask.resize(img.size, Image.LANCZOS)
 
         return [mask]