Update app.py

app.py

@@ -93,7 +93,7 @@ def create_video(frames, fps, type):
     return type + "_result.mp4"
 
     
-def infer(url_in,interpolation,fps_output,resize_n,winsize):
+def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
     
     fps_output = logscale(fps_output)
     # 1. break video into frames and get FPS
@@ -137,19 +137,27 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize):
     prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
     hsv = np.zeros_like(fr1)
     hsv[..., 1] = 255
-    res = np.zeros_like(prvs)
-    flow = res
+    if o_flow == True:
+      res = np.zeros_like(prvs)
+      flow = res
+    else:
+      res = prvs
+        
     while(depth_map.isOpened()):
       ret, fr2 = depth_map.read()
       if ret == False:
         break
       nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
-      fl = cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, winsize, 3, 5, 1.2, 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)
-      rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
-      rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
+      if o_flow == True:
+        fl = cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, winsize, 3, 5, 1.2, 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)
+        rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
+        rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
+      else:
+        rgb = nxt
+          
       alpha = 1.0/len(frames_list)
       beta = (1.0 - alpha)
       res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
@@ -196,6 +204,7 @@ with gr.Blocks() as demo:
             with gr.Column():
                 url_input = gr.Textbox(value="./examples/streetview.mp4", label="URL")
                 resize_num = gr.Slider(minimum=1, maximum=4096, step=1, value=256, label="Resize to width: ")
+                of_check = gr.Checkbox(value=False, label="Detect motion for depth map: ")
                 winsize_num = gr.Slider(minimum=1, maximum=256, step=1, value=15, label="Motion detection window size: ")
                 with gr.Row():
                     interpolation_slider = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Interpolation Steps: ")
@@ -213,13 +222,13 @@ with gr.Blocks() as demo:
                 depth_output = gr.ImageEditor(image_mode="L", interactive=True, label="Depth map")
     
     gr.Examples(
-        examples=[["./examples/streetview.mp4", 1, 0, 256, 15]],
+        examples=[["./examples/streetview.mp4", 1, 0, 256, 15, False]],
         fn=infer,
-        inputs=[url_input,interpolation_slider,fps_output_slider,resize_num,winsize_num],
+        inputs=[url_input,interpolation_slider,fps_output_slider,resize_num,winsize_num,of_check],
         outputs=[video_output,file_output,depth_output],
         cache_examples=True
     )
     
-    submit_btn.click(fn=infer, inputs=[url_input,interpolation_slider,fps_output_slider,resize_num,winsize_num], 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])
 
 demo.launch()