From af8c8e2f0cc9deca3f3d7b86f85e2301fadd6109 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Tue, 14 Jun 2022 10:13:54 +0100 Subject: [PATCH] Fix minor typo in optical flow gallery example (#6163) --- gallery/plot_optical_flow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gallery/plot_optical_flow.py b/gallery/plot_optical_flow.py index 9e8d0006f1a..b0a93209877 100644 --- a/gallery/plot_optical_flow.py +++ b/gallery/plot_optical_flow.py @@ -180,7 +180,7 @@ def preprocess(img1_batch, img2_batch): # # Note: it would be faster to predict batches of flows instead of individual flows # 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] # flow_img = flow_to_image(predicted_flow).to("cpu") # output_folder = "/tmp/" # Update this to the folder of your choice