Prechádzať zdrojové kódy

remove resampling from session_simple

Sami Ede 2 rokov pred
rodič
commit
79ab2b0ad0
1 zmenil súbory, kde vykonal 1 pridanie a 1 odobranie
  1. 1 1
      rembg/session_simple.py

+ 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]