Update app.py
Browse files
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 =
|
209 |
-
d =
|
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
|
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))
|