Update app.py
Browse files
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 |
-
|
106 |
-
|
107 |
-
|
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"
|