video_frame_interpolation

Running

freealise commited on Dec 1, 2024

Commit

647055f

verified ·

1 Parent(s): dd5a8d5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -175,7 +175,7 @@ def remove_bg(fl, count):
     if count % 2: # odd: is photo without the flash
         fr = cv2.imread(fl_).astype(np.uint8)
-        fr = fr_diff - fr
         cv2.imwrite(fl_, fr)
     else: # even: with the flash
         fl_ = fl.split(".")[0] + "_.png"

     if count % 2: # odd: is photo without the flash
         fr = cv2.imread(fl_).astype(np.uint8)
+        fr = fr_diff - cv2.cvtColor(fr, cv2.COLOR_BGR2GRAY)
         cv2.imwrite(fl_, fr)
     else: # even: with the flash
         fl_ = fl.split(".")[0] + "_.png"