Update app.py
Browse files
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(
|
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)
|