Update app.py
Browse files
app.py
CHANGED
|
@@ -206,8 +206,8 @@ def infer():
|
|
| 206 |
# display the PIL image
|
| 207 |
#img.show()
|
| 208 |
img.save('frame_input.jpg')
|
| 209 |
-
|
| 210 |
-
|
| 211 |
return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
|
| 212 |
####################################
|
| 213 |
# Bonus: Creating GIFs of predicted flows
|
|
|
|
| 206 |
# display the PIL image
|
| 207 |
#img.show()
|
| 208 |
img.save('frame_input.jpg')
|
| 209 |
+
res = get_warp_res("frame_input.jpg", "flofile.flo", fname_output='warped.png')
|
| 210 |
+
print(res)
|
| 211 |
return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
|
| 212 |
####################################
|
| 213 |
# Bonus: Creating GIFs of predicted flows
|