Update app.py

app.py

@@ -129,15 +129,22 @@ def infer(url_in,interpolation,fps_output):
     depth_map = cv2.VideoCapture(final_vid)
 
     ret, fr1 = depth_map.read()
-    prvs = cv2.cvtColor(fr1, cv2.COLOR_BGR2GRAY)
+    prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
+    hsv = np.zeros_like(fr1)
+    hsv[..., 1] = 255
     res = prvs
     flow = res
     while(depth_map.isOpened()):
       ret, fr2 = depth_map.read()
       if ret == False:
         break
-      nxt = cv2.cvtColor(fr2, cv2.COLOR_BGR2GRAY)
+      nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
       cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, 15, 3, 5, 1.2, 0)
+      mag, ang = cv2.cartToPolar(flow[..., 0], flow[..., 1])
+      hsv[..., 0] = ang*180/np.pi/2
+      hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
+      rgba = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGBA)
+      cv2.imshow('fr2', rgba)
       alpha = 1.0/len(frames_list)
       beta = (1.0 - alpha)
       res = cv2.addWeighted(flow, alpha, res, beta, 0.0)