Bladeren bron

fix: Expected RGB image of shape (?, ?, 3), but image.shape is (?, ?, 4).

goecho 3 jaren geleden
bovenliggende
commit
fb3b9feb72
1 gewijzigde bestanden met toevoegingen van 4 en 0 verwijderingen
  1. 4 0
      rembg/bg.py

+ 4 - 0
rembg/bg.py

@@ -37,6 +37,10 @@ def alpha_matting_cutout(
     background_threshold: int,
     erode_structure_size: int,
 ) -> PILImage:
+
+    if img.mode == "RGBA" or img.mode == "CMYK":
+        img = img.convert("RGB")
+
     img = np.asarray(img)
     mask = np.asarray(mask)