freealise commited on
Commit
bc128a0
·
verified ·
1 Parent(s): adc979c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -205,8 +205,8 @@ def remove_bg(fl, count, mh, ms, md, lm):
205
 
206
  cv2.imwrite(fl, fr)
207
 
208
- b = 15
209
- d = 1
210
  element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
211
  mask_e = cv2.erode(mask, element) * 255
212
  element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * d + 1, 2 * d + 1), (d, d))
@@ -218,14 +218,13 @@ def remove_bg(fl, count, mh, ms, md, lm):
218
  a_map = cv2.imread(f"{str(count)}_cutout.png", cv2.IMREAD_UNCHANGED).astype(np.uint8)
219
  B, G, R, A = cv2.split(a_map)
220
  alpha = A / 255
221
-
222
  R = (255 * (1 - alpha) + R * alpha).astype(np.uint8)
223
  G = (255 * (1 - alpha) + G * alpha).astype(np.uint8)
224
  B = (255 * (1 - alpha) + B * alpha).astype(np.uint8)
225
  fr = cv2.merge((B, G, R))
226
 
227
  cv2.imwrite(fl, fr)
228
- return fl #f"{str(count)}_cutout.png"
229
  else: # even: with the flash
230
  fl_ = fl.split(".")[0] + "_.png"
231
  cv2.imwrite(fl_, fr_diff.astype(np.uint8))
 
205
 
206
  cv2.imwrite(fl, fr)
207
 
208
+ b = 1
209
+ d = 15
210
  element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
211
  mask_e = cv2.erode(mask, element) * 255
212
  element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * d + 1, 2 * d + 1), (d, d))
 
218
  a_map = cv2.imread(f"{str(count)}_cutout.png", cv2.IMREAD_UNCHANGED).astype(np.uint8)
219
  B, G, R, A = cv2.split(a_map)
220
  alpha = A / 255
 
221
  R = (255 * (1 - alpha) + R * alpha).astype(np.uint8)
222
  G = (255 * (1 - alpha) + G * alpha).astype(np.uint8)
223
  B = (255 * (1 - alpha) + B * alpha).astype(np.uint8)
224
  fr = cv2.merge((B, G, R))
225
 
226
  cv2.imwrite(fl, fr)
227
+ return fl
228
  else: # even: with the flash
229
  fl_ = fl.split(".")[0] + "_.png"
230
  cv2.imwrite(fl_, fr_diff.astype(np.uint8))