fffiloni commited on
Commit
4a10134
·
1 Parent(s): 6b9ae2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -102,9 +102,9 @@ def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
102
  print(f"FNAME IMAGE: {fname_image}")
103
  im2 = imread(fname_image)
104
  print(f"FNAME IMAGE READED: {im2}")
105
- #flow = readFlowFile(fname_flow)
106
- #im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
107
- #imwrite(fname_output, im_warped)
108
 
109
  def infer():
110
  video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"
 
102
  print(f"FNAME IMAGE: {fname_image}")
103
  im2 = imread(fname_image)
104
  print(f"FNAME IMAGE READED: {im2}")
105
+ flow = readFlowFile(fname_flow)
106
+ im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
107
+ imwrite(fname_output, im_warped)
108
 
109
  def infer():
110
  video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"