Update app.py
Browse files
app.py
CHANGED
|
@@ -109,7 +109,7 @@ def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
|
|
| 109 |
print(f"FNAME IMAGE READED: {im2.shape}")
|
| 110 |
flow = fname_flow.cpu().detach().numpy()
|
| 111 |
print(f"FNAME FLOW READED: {flow.shape}")
|
| 112 |
-
|
| 113 |
#imwrite(fname_output, im_warped)
|
| 114 |
|
| 115 |
def infer():
|
|
|
|
| 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():
|