Update app.py

app.py

@@ -149,13 +149,13 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
       if ret == False:
         if o_flow == False:
           rgb = prvs
-          alpha = 1.0/len(result_frames)
+          alpha = 1.0/(i+1)
           beta = (1.0 - alpha)
           res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
         break
       nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
       if o_flow == True:
-        fl = cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, winsize, 3, 5, 1.2, 0)
+        fl = cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 7, winsize, 7, 7, 1.5, 0)
         mag, ang = cv2.cartToPolar(fl[..., 0], fl[..., 1])
         hsv[..., 0] = ang*180/np.pi/2
         hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
@@ -164,7 +164,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
       else:
         rgb = prvs
 
-      alpha = 1.0/len(result_frames)
+      alpha = 1.0/(i+1)
       beta = (1.0 - alpha)
       res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
       rgb = cv2.cvtColor(rgb, cv2.COLOR_GRAY2RGB)
@@ -188,7 +188,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
     depth_map.release()
     cv2.destroyAllWindows()
 
-    return final_vid, files, depth_vid
+    return final_vid, files, depth_vid, 'opticalfb.png'
 
 
 def logscale(linear):
@@ -249,15 +249,16 @@ with gr.Blocks() as demo:
                 video_output = gr.Video()
                 file_output = gr.File()
                 depth_output = gr.Video()
+                depth_avg = gr.Image()
     
     gr.Examples(
-        examples=[["./examples/man-in-museum.mp4", 2, 0, 256, 15, True]],
+        examples=[["./examples/man-in-museum-reverse-cut.mp4", 2, 0, 256, 15, True]],
         fn=infer,
-        inputs=[url_input,interpolation_slider,fps_output_slider,resize_num,winsize_num,of_check],
-        outputs=[video_output,file_output,depth_output],
+        inputs=[url_input, interpolation_slider, fps_output_slider, resize_num, winsize_num, of_check],
+        outputs=[video_output, file_output, depth_output, depth_avg],
         cache_examples=True
     )
     
-    submit_btn.click(fn=infer, inputs=[url_input,interpolation_slider,fps_output_slider,resize_num,winsize_num,of_check], outputs=[video_output, file_output, depth_output])
+    submit_btn.click(fn=infer, inputs=[url_input, interpolation_slider, fps_output_slider, resize_num, winsize_num, of_check], outputs=[video_output, file_output, depth_output, depth_avg])
 
 demo.launch()