fffiloni commited on
Commit
4cfe8b0
·
1 Parent(s): 8e586dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)