Update app.py
Browse files
app.py
CHANGED
@@ -66,6 +66,7 @@ def warp_flow(img, flow):
|
|
66 |
|
67 |
|
68 |
def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
|
|
|
69 |
flow21 = np.load(flo_path, allow_pickle=True)
|
70 |
frame1pil = np.array(frame1.convert('RGB').resize((flow21.shape[1],flow21.shape[0])))
|
71 |
frame1_warped21 = warp_flow(frame1pil, flow21)
|
|
|
66 |
|
67 |
|
68 |
def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
|
69 |
+
flo_path = flo_path.cpu().data.numpy()
|
70 |
flow21 = np.load(flo_path, allow_pickle=True)
|
71 |
frame1pil = np.array(frame1.convert('RGB').resize((flow21.shape[1],flow21.shape[0])))
|
72 |
frame1_warped21 = warp_flow(frame1pil, flow21)
|