fffiloni commited on
Commit
42ba5b2
·
1 Parent(s): c36084d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -106,10 +106,11 @@ def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
106
  print(f"FNAME IMAGE: {fname_image}")
107
  im2 = imread(fname_image)
108
  #im2 = fname_image
109
- print(f"FNAME IMAGE READED: {im2}")
110
  flow = fname_flow.cpu().detach().numpy()
111
- im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
112
- imwrite(fname_output, im_warped)
 
113
 
114
  def infer():
115
  video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"
 
106
  print(f"FNAME IMAGE: {fname_image}")
107
  im2 = imread(fname_image)
108
  #im2 = fname_image
109
+ print(f"FNAME IMAGE READED: {im2.shape}")
110
  flow = fname_flow.cpu().detach().numpy()
111
+ print(f"FNAME FLOW READED: {flow.shape}")
112
+ #im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
113
+ #imwrite(fname_output, im_warped)
114
 
115
  def infer():
116
  video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"