Update app.py
Browse files
app.py
CHANGED
@@ -176,13 +176,12 @@ def remove_bg(fl, count):
|
|
176 |
if count % 2: # odd: is photo without the flash
|
177 |
fr = cv2.imread(fl_).astype(np.uint8)
|
178 |
fr = fr_diff - cv2.cvtColor(fr, cv2.COLOR_BGR2GRAY)
|
179 |
-
cv2.imwrite(
|
|
|
180 |
else: # even: with the flash
|
181 |
fl_ = fl.split(".")[0] + "_.png"
|
182 |
-
print(fl_)
|
183 |
cv2.imwrite(fl_, fr_diff.astype(np.uint8))
|
184 |
-
|
185 |
-
return fl_
|
186 |
|
187 |
def sharpest(fl, i):
|
188 |
break_vid = get_frames(fl, "vid_input_frame", "origin", i)
|
|
|
176 |
if count % 2: # odd: is photo without the flash
|
177 |
fr = cv2.imread(fl_).astype(np.uint8)
|
178 |
fr = fr_diff - cv2.cvtColor(fr, cv2.COLOR_BGR2GRAY)
|
179 |
+
cv2.imwrite(fl, fr)
|
180 |
+
return fl
|
181 |
else: # even: with the flash
|
182 |
fl_ = fl.split(".")[0] + "_.png"
|
|
|
183 |
cv2.imwrite(fl_, fr_diff.astype(np.uint8))
|
184 |
+
return fl_
|
|
|
185 |
|
186 |
def sharpest(fl, i):
|
187 |
break_vid = get_frames(fl, "vid_input_frame", "origin", i)
|