freealise commited on
Commit
bb6622d
1 Parent(s): 656ebf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -143,8 +143,8 @@ def infer(url_in,interpolation,fps_output):
143
  mag, ang = cv2.cartToPolar(fl[..., 0], fl[..., 1])
144
  hsv[..., 0] = ang*180/np.pi/2
145
  hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
146
- rgba = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGBA)
147
- cv2.imshow('fr2', rgba)
148
  alpha = 1.0/len(frames_list)
149
  beta = (1.0 - alpha)
150
  res = cv2.addWeighted(flow, alpha, res, beta, 0.0)
 
143
  mag, ang = cv2.cartToPolar(fl[..., 0], fl[..., 1])
144
  hsv[..., 0] = ang*180/np.pi/2
145
  hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
146
+ rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
147
+ cv2.imshow('fr2', rgb)
148
  alpha = 1.0/len(frames_list)
149
  beta = (1.0 - alpha)
150
  res = cv2.addWeighted(flow, alpha, res, beta, 0.0)