a demo for testing on a pair of images

Author: fitsumreda

run_a_pair.py

@@ -10,6 +10,7 @@
     parser = argparse.ArgumentParser()
     parser.add_argument('--fp16', action='store_true', help='Run model in pseudo-fp16 mode (fp16 storage fp32 math).')
     parser.add_argument("--rgb_max", type=float, default=255.)
+    
     args = parser.parse_args()
 
     # initial a Net
@@ -41,4 +42,4 @@ def writeFlow(name, flow):
 
 
     data = result.data.cpu().numpy().transpose(1, 2, 0)
-    writeFlow("/home/hjj/flownet2-master/data/FlyingChairs_examples/0000007-img.flo", data)
+    writeFlow("/home/hjj/flownet2-master/data/FlyingChairs_examples/0000007-img.flo", data)