Spaces:

vivym
/

image-matting-app

Runtime error

vivym commited on Nov 25, 2022

Commit

6655549

•

1 Parent(s): 47bf4d4

bugfix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -70,7 +70,7 @@ def image_matting(
     alpha = (alpha * 255).astype(np.uint8)
     kernel = np.ones((5, 5), np.uint8)
-    if morph_op == "dilate":
         alpha = cv2.dilate(alpha, kernel, iterations=int(morph_op_factor))
     else:
         alpha = cv2.erode(alpha, kernel, iterations=int(morph_op_factor))

     alpha = (alpha * 255).astype(np.uint8)
     kernel = np.ones((5, 5), np.uint8)
+    if morph_op == "Dilate":
         alpha = cv2.dilate(alpha, kernel, iterations=int(morph_op_factor))
     else:
         alpha = cv2.erode(alpha, kernel, iterations=int(morph_op_factor))