Update app.py

app.py

@@ -112,24 +112,21 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
     # 2. prepare frames result arrays
     result_frames = []
     print("set stop frames to: " + str(n_frame))
-    
+
+    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)
-            rgb_next = cv2.imread(next_frame).astype(np.uint8)
-
-            diff = np.abs(rgb_next - rgb)
-            lo = np.array([0,0,0])
-            hi = np.array([16,16,16])
-            mask = cv2.inRange(diff, lo, hi)
-            rgb[mask>0] = (255,255,255)
-            rgb_next[mask>0] = (255,255,255)
-
-            cv2.imwrite(frame, rgb)
-            cv2.imwrite(next_frame, rgb_next)
-            
             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])