Update app.py

app.py

@@ -115,18 +115,23 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
 
     bg = cv2.imread(frames_list[len(frames_list) - 1]).astype(np.uint8)
     for idx, frame in enumerate(frames_list[0:int(n_frame)]):
-
-        rgb = cv2.imread(frame).astype(np.uint8)
-        diff = np.abs(rgb - bg)
-        lo = np.array([0,0,0])
-        hi = np.array([4,4,4])
-        mask = cv2.inRange(diff, lo, hi)
-        rgb[mask>0] = (255,255,255)
-        cv2.imwrite(frame, rgb)
-        
         if idx < len(frames_list) - 1:
             next_frame = frames_list[idx+1]
 
+            rgb = cv2.imread(frame).astype(np.uint8)
+            next_rgb = cv2.imread(next_frame).astype(np.uint8)
+            diff_f = np.abs(rgb - next_rgb)
+            diff = np.abs(rgb - bg)
+            
+            lo = np.array([0,0,0])
+            hi = np.array([16,16,16])
+            mask_f = cv2.inRange(diff_f, lo, hi)
+            mask = cv2.inRange(diff, lo, hi)
+            rgb[mask>0 and mask_f==0] = (255,255,255)
+            next_rgb[mask>0 and mask_f==0] = (255,255,255)
+            cv2.imwrite(frame, rgb)
+            cv2.imwrite(next_frame, next_rgb)
+
             interpolated_frames = do_interpolation(frame, next_frame, interpolation) # should return a list of 3 interpolated frames
             break_interpolated_video = get_frames(interpolated_frames, "interpol", f"{idx}_", 0)
             print(break_interpolated_video[0])