浏览代码

add mask post process

Daniel Gatis 3 年之前
父节点
当前提交
37359b3079
共有 1 个文件被更改,包括 2 次插入8 次删除
  1. 2 8
      rembg/bg.py

+ 2 - 8
rembg/bg.py

@@ -3,15 +3,9 @@ from enum import Enum
 from typing import List, Optional, Union
 
 import numpy as np
+from cv2 import (BORDER_DEFAULT, MORPH_ELLIPSE, MORPH_OPEN, GaussianBlur,
+                 getStructuringElement, morphologyEx)
 from PIL import Image
-from cv2 import (
-    getStructuringElement,
-    morphologyEx,
-    GaussianBlur,
-    MORPH_OPEN,
-    MORPH_ELLIPSE,
-    BORDER_DEFAULT,
-)
 from PIL.Image import Image as PILImage
 from pymatting.alpha.estimate_alpha_cf import estimate_alpha_cf
 from pymatting.foreground.estimate_foreground_ml import estimate_foreground_ml