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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -139,8 +139,8 @@ def infer(url_in,interpolation,fps_output):
139
  if ret == False:
140
  break
141
  nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
142
- cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, 15, 3, 5, 1.2, 0)
143
- mag, ang = cv2.cartToPolar(flow[..., 0], flow[..., 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)
 
139
  if ret == False:
140
  break
141
  nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
142
+ fl = cv2.calcOpticalFlowFarneback(prvs, nxt, flow, 0.5, 3, 15, 3, 5, 1.2, 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
  rgba = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGBA)