Update app.py
Browse files
app.py
CHANGED
@@ -88,7 +88,7 @@ def read_flo_file(filename):
|
|
88 |
return data2d
|
89 |
|
90 |
def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
|
91 |
-
|
92 |
flow21 = np.array(flo_path)
|
93 |
frame1pil = np.array(frame1)
|
94 |
frame1_warped21 = warp_flow(frame1pil, flow21)
|
|
|
88 |
return data2d
|
89 |
|
90 |
def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
|
91 |
+
flo_path = flo_path.cpu().data.numpy()
|
92 |
flow21 = np.array(flo_path)
|
93 |
frame1pil = np.array(frame1)
|
94 |
frame1_warped21 = warp_flow(frame1pil, flow21)
|