Update app.py

app.py

@@ -118,21 +118,6 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
         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)
-            m = cv2.bitwise_and(mask_f, mask)
-            rgb[m>0] = (255,255,255)
-            next_rgb[m>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])
@@ -183,10 +168,18 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
       alpha = 1.0/len(result_frames)
       beta = (1.0 - alpha)
       res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
-      rgba = cv2.cvtColor(rgb, cv2.COLOR_GRAY2RGBA)
-        
-      cv2.imwrite(f"opticalfb{i}.png", rgba)
+      rgb = cv2.cvtColor(rgb, cv2.COLOR_GRAY2RGB)
+
+      lo = np.array([0,0,0])
+      hi = np.array([16,16,16])
+      mask = cv2.inRange(rgb, lo, hi)
+      #m = cv2.bitwise_and(mask_f, mask)
+      #diff = np.abs(rgba - bg)
+      fr1[mask>0] = (255,255,255,255)
+
+      cv2.imwrite(f"opticalfb{i}.png", fr1)
       depth_frames.append(f"opticalfb{i}.png")
+        
       i+=1
       prvs = nxt