Update app.py
Browse files
app.py
CHANGED
@@ -71,8 +71,8 @@ def warp_flow(img, flow, mul=1.):
|
|
71 |
#flow = np.load(flow)
|
72 |
h, w = flow.shape[:2]
|
73 |
flow = flow.copy()
|
74 |
-
flow[:, :, 0]
|
75 |
-
flow[:, :, 1]
|
76 |
# print('flow stats', flow.max(), flow.min(), flow.mean())
|
77 |
# print(flow)
|
78 |
flow*=mul
|
|
|
71 |
#flow = np.load(flow)
|
72 |
h, w = flow.shape[:2]
|
73 |
flow = flow.copy()
|
74 |
+
flow[:, :, 0] + np.arange(w)
|
75 |
+
flow[:, :, 1] + np.arange(h)[:, np.newaxis]
|
76 |
# print('flow stats', flow.max(), flow.min(), flow.mean())
|
77 |
# print(flow)
|
78 |
flow*=mul
|