Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 5, 2023

Commit

7a51391

1 Parent(s): 2e90b3c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -226,7 +226,7 @@ def infer():
     # print('flow stats mul', flow.max(), flow.min(), flow.mean())
     # res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
     res = cv2.remap(frame1pil, numpy_array_flow, None, cv2.INTER_LANCZOS4)
-    #print(res)
     return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
 ####################################
 # Bonus: Creating GIFs of predicted flows

     # print('flow stats mul', flow.max(), flow.min(), flow.mean())
     # res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
     res = cv2.remap(frame1pil, numpy_array_flow, None, cv2.INTER_LANCZOS4)
+    print(res)
     return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
 ####################################
 # Bonus: Creating GIFs of predicted flows