fix alpha matting

danielgatis · Apr 18, 2022 · 92e768c · 92e768c
1 parent f72fe5f
commit 92e768c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rembg/bg.py b/rembg/bg.py
@@ -90,7 +90,10 @@ def remove(
     if session is None:
         session = ort_session("u2net")
 
-    mask = predict(session, np.array(img.convert("RGB"))).convert("L")
+    img = img.convert("RGB")
+
+    mask = predict(session, np.array(img))
+    mask = mask.convert("L")
     mask = mask.resize(img.size, Image.LANCZOS)
 
     if only_mask: