Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 5, 2023

Commit

ea0a6d1

•

1 Parent(s): 2d673d7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,7 +104,7 @@ def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
     print(f"FNAME IMAGE: {fname_image}")
     im2 = imread(fname_image)
     print(f"FNAME IMAGE READED: {im2}")
-    flow = fname_flow
     im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
     imwrite(fname_output, im_warped)

     print(f"FNAME IMAGE: {fname_image}")
     im2 = imread(fname_image)
     print(f"FNAME IMAGE READED: {im2}")
+    flow = fname_flow.cpu()
     im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
     imwrite(fname_output, im_warped)