Unverified Commit 3c94a490 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix minor typo in optical flow gallery example (#6163)

parent 9430be76
...@@ -180,7 +180,7 @@ plot(grid) ...@@ -180,7 +180,7 @@ plot(grid)
# # Note: it would be faster to predict batches of flows instead of individual flows # # Note: it would be faster to predict batches of flows instead of individual flows
# img1, img2 = preprocess(img1, img2) # img1, img2 = preprocess(img1, img2)
# list_of_flows = model(img1.to(device), img1.to(device)) # list_of_flows = model(img1.to(device), img2.to(device))
# predicted_flow = list_of_flows[-1][0] # predicted_flow = list_of_flows[-1][0]
# flow_img = flow_to_image(predicted_flow).to("cpu") # flow_img = flow_to_image(predicted_flow).to("cpu")
# output_folder = "/tmp/" # Update this to the folder of your choice # output_folder = "/tmp/" # Update this to the folder of your choice
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment