vivym commited on
Commit
6655549
1 Parent(s): 47bf4d4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def image_matting(
70
 
71
  alpha = (alpha * 255).astype(np.uint8)
72
  kernel = np.ones((5, 5), np.uint8)
73
- if morph_op == "dilate":
74
  alpha = cv2.dilate(alpha, kernel, iterations=int(morph_op_factor))
75
  else:
76
  alpha = cv2.erode(alpha, kernel, iterations=int(morph_op_factor))
 
70
 
71
  alpha = (alpha * 255).astype(np.uint8)
72
  kernel = np.ones((5, 5), np.uint8)
73
+ if morph_op == "Dilate":
74
  alpha = cv2.dilate(alpha, kernel, iterations=int(morph_op_factor))
75
  else:
76
  alpha = cv2.erode(alpha, kernel, iterations=int(morph_op_factor))