Update app.py

app.py

@@ -211,16 +211,10 @@ def infer():
     print(f"img1_batch dtype: {img1_batch.dtype}")
     #res = get_warp_res(img1_batch, "predicted_flow.jpg", 'warped.png')
 
-    h, w = predicted_flow.shape[:2]
-    flow = predicted_flow.copy()
-    flow[:, :, 0] + np.arange(w)
-    flow[:, :, 1] + np.arange(h)[:, np.newaxis]
-    # print('flow stats', flow.max(), flow.min(), flow.mean())
-    # print(flow)
-    flow*1.
-    # print('flow stats mul', flow.max(), flow.min(), flow.mean())
-    # res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
-    res = cv2.remap(img1_batch, flow, None, cv2.INTER_LANCZOS4)
+    numpy_array_flow = predicted_flow.numpy()
+    print(f"numpy_array_flow: {numpy_array_flow}")
+    print(f"numpy_array_flow shape: {numpy_array_flow.shape}")
+    print(f"numpy_array_flow dtype: {numpy_array_flow.dtype}")
     
     #print(res)
     return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'