fffiloni commited on
Commit
62a40a6
1 Parent(s): 229c22f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -61,6 +61,9 @@ def write_flo(flow, filename):
61
  #warp using scipy
62
  def warp_image(im, flow):
63
  print(f"IMAGE: {im}")
 
 
 
64
  print(f"FLOW AV: {flow}")
65
  flow = flow.cpu().data.numpy()
66
  flow = flow.astype(np.float32)
 
61
  #warp using scipy
62
  def warp_image(im, flow):
63
  print(f"IMAGE: {im}")
64
+ im = im.cpu().data.numpy()
65
+ im = im.astype(np.float32)
66
+ print(f"IMAGE AP: {im}")
67
  print(f"FLOW AV: {flow}")
68
  flow = flow.cpu().data.numpy()
69
  flow = flow.astype(np.float32)