Update app.py
Browse files
app.py
CHANGED
@@ -61,6 +61,9 @@ def write_flo(flow, filename):
|
|
61 |
#warp using scipy
|
62 |
def warp_image(im, flow):
|
63 |
print(f"IMAGE: {im}")
|
|
|
|
|
|
|
64 |
print(f"FLOW AV: {flow}")
|
65 |
flow = flow.cpu().data.numpy()
|
66 |
flow = flow.astype(np.float32)
|
|
|
61 |
#warp using scipy
|
62 |
def warp_image(im, flow):
|
63 |
print(f"IMAGE: {im}")
|
64 |
+
im = im.cpu().data.numpy()
|
65 |
+
im = im.astype(np.float32)
|
66 |
+
print(f"IMAGE AP: {im}")
|
67 |
print(f"FLOW AV: {flow}")
|
68 |
flow = flow.cpu().data.numpy()
|
69 |
flow = flow.astype(np.float32)
|