Daniel Gatis 1 سال پیش
والد
کامیت
49d1686f65
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      rembg/bg.py

+ 1 - 1
rembg/bg.py

@@ -157,7 +157,7 @@ def post_process(mask: np.ndarray) -> np.ndarray:
     """
     mask = morphologyEx(mask, MORPH_OPEN, kernel)
     mask = GaussianBlur(mask, (5, 5), sigmaX=2, sigmaY=2, borderType=BORDER_DEFAULT)
-    mask = np.where(mask < 127, 0, 255).astype(np.uint8)  # convert again to binary
+    mask = np.where(mask < 127, 0, 255).astype(np.uint8)  # type: ignore
     return mask