浏览代码

fix linters

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