Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 5, 2023

Commit

c0f109a

1 Parent(s): c5595d9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,7 +57,7 @@ def write_flo(flow, filename):
     f.close()
 def warp_flow(img, flow):
-      flow = flow.cpu().data.numpy()
       h, w = flow.shape[:2]
       flow = flow.copy()
       flow[:, :, 0] += np.arange(w)

     f.close()
 def warp_flow(img, flow):
+      #flow = flow.cpu().data.numpy()
       h, w = flow.shape[:2]
       flow = flow.copy()
       flow[:, :, 0] += np.arange(w)