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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -60,8 +60,11 @@ def write_flo(flow, filename):
60
 
61
  #warp using scipy
62
  def warp_image(im, flow):
 
 
63
  flow = flow.cpu().data.numpy()
64
  flow = flow.astype(np.float32)
 
65
  h, w = flow.shape[:2]
66
  flow[:,:,0] += np.arange(w)
67
  flow[:,:,1] += np.arange(h)[:,np.newaxis]
 
60
 
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)
67
+ print(f"FLOW AP: {flow}")
68
  h, w = flow.shape[:2]
69
  flow[:,:,0] += np.arange(w)
70
  flow[:,:,1] += np.arange(h)[:,np.newaxis]