diff options
-rw-r--r-- | test/test_warp.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_warp.lua b/test/test_warp.lua index 68a4ac1..5c4a1ed 100644 --- a/test/test_warp.lua +++ b/test/test_warp.lua @@ -32,8 +32,8 @@ flow_scale[1] = grid_y flow_scale[2] = grid_x flow_scale[1]:add(1):mul(0.5) -- 0 to 1 flow_scale[2]:add(1):mul(0.5) -- 0 to 1 -flow_scale[1]:mul(height) -flow_scale[2]:mul(width) +flow_scale[1]:mul(height-1) +flow_scale[2]:mul(width-1) flow:add(flow_scale) t0 = sys.clock() @@ -86,8 +86,8 @@ flow_scale[1] = grid_y flow_scale[2] = grid_x flow_scale[1]:add(1):mul(0.5) -- 0 to 1 flow_scale[2]:add(1):mul(0.5) -- 0 to 1 -flow_scale[1]:mul(height) -flow_scale[2]:mul(width) +flow_scale[1]:mul(height-1) +flow_scale[2]:mul(width-1) flow:add(flow_scale) flow_rot = torch.FloatTensor() |