Update app.py

app.py

@@ -140,7 +140,8 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
     hsv[..., 1] = 255
     res = np.zeros_like(prvs)
     flow = res
-        
+
+    i=0
     while(depth_map.isOpened()):
       ret, fr2 = depth_map.read()
       if ret == False:
@@ -149,7 +150,6 @@ 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)
-          depth_frames.append(rgb)
         break
       nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
       if o_flow == True:
@@ -165,7 +165,9 @@ 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)
-      depth_frames.append(rgb)
+      cv2.imwrite(f"opticalfb{str(i)}.png", rgb)
+      depth_frames.append(f"opticalfb{str(i)}.png")
+      i+=1
       prvs = nxt
 
     print("averaged frames: " + str(len(result_frames)))