Update app.py
Browse files
app.py
CHANGED
@@ -210,12 +210,12 @@ def infer():
|
|
210 |
transform = T.ToPILImage()
|
211 |
|
212 |
# convert the tensor to PIL image using above transform
|
213 |
-
img = transform(
|
214 |
|
215 |
# display the PIL image
|
216 |
#img.show()
|
217 |
img.save('frame_input.jpg')
|
218 |
-
res = get_warp_res(
|
219 |
#print(res)
|
220 |
return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
|
221 |
####################################
|
|
|
210 |
transform = T.ToPILImage()
|
211 |
|
212 |
# convert the tensor to PIL image using above transform
|
213 |
+
img = transform(frames[100])
|
214 |
|
215 |
# display the PIL image
|
216 |
#img.show()
|
217 |
img.save('frame_input.jpg')
|
218 |
+
res = get_warp_res(frames[100], predicted_flow, fname_output='warped.png')
|
219 |
#print(res)
|
220 |
return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
|
221 |
####################################
|