Update app.py
Browse files
app.py
CHANGED
|
@@ -57,6 +57,7 @@ def write_flo(flow, filename):
|
|
| 57 |
f.close()
|
| 58 |
|
| 59 |
def warp_flow(img, flow):
|
|
|
|
| 60 |
h, w = flow.shape[:2]
|
| 61 |
flow = flow.copy()
|
| 62 |
flow[:, :, 0] += np.arange(w)
|
|
|
|
| 57 |
f.close()
|
| 58 |
|
| 59 |
def warp_flow(img, flow):
|
| 60 |
+
flow = flow.cpu().data.numpy()
|
| 61 |
h, w = flow.shape[:2]
|
| 62 |
flow = flow.copy()
|
| 63 |
flow[:, :, 0] += np.arange(w)
|