Update app.py

app.py

@@ -132,7 +132,7 @@ def infer(url_in,interpolation,fps_output):
     prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
     hsv = np.zeros_like(fr1)
     hsv[..., 1] = 255
-    res = prvs
+    res = hsv
     flow = res
     while(depth_map.isOpened()):
       ret, fr2 = depth_map.read()
@@ -144,9 +144,10 @@ def infer(url_in,interpolation,fps_output):
       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)
       alpha = 1.0/len(frames_list)
       beta = (1.0 - alpha)
-      res = cv2.addWeighted(flow, alpha, res, beta, 0.0)
+      res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
       prvs = nxt
 
     cv2.imwrite('opticalfb.png', res)