add KITTI example

liruoteng · Oct 30, 2017 · fcf81cc · fcf81cc
1 parent 5185cf1
commit fcf81cc
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 0 deletions.
diff --git a/data/example/KITTI/flow_gt.png b/data/example/KITTI/flow_gt.png
diff --git a/data/example/KITTI/frame1.png b/data/example/KITTI/frame1.png
diff --git a/data/example/KITTI/frame2.png b/data/example/KITTI/frame2.png
diff --git a/lib/flowlib.py b/lib/flowlib.py
@@ -376,6 +376,8 @@ def warp_image(im, flow):
     n = image_height * image_width
     (iy, ix) = np.mgrid[0:image_height, 0:image_width]
     (fy, fx) = np.mgrid[0:flow_height, 0:flow_width]
+    fx = fx.astype(np.float64)
+    fy = fy.astype(np.float64)
     fx += flow[:,:,0]
     fy += flow[:,:,1]
     mask = np.logical_or(fx <0 , fx > flow_width)