diff options
| author | Michael 'myrhev' Mathieu <michael.mathieu@ens.fr> | 2012-03-10 05:09:24 +0400 |
|---|---|---|
| committer | Michael 'myrhev' Mathieu <michael.mathieu@ens.fr> | 2012-03-10 05:09:24 +0400 |
| commit | 90503963e940a592205f66a704b3d1e96e894c2c (patch) | |
| tree | b40fdbc1c4a6843f727f3b888b9558aff7f21e3e /SpatialMatching.lua | |
| parent | c21d098655ea9732f60a372fa86710cb8669862a (diff) | |
Fix wrong output dimension in SpatialMatching
Diffstat (limited to 'SpatialMatching.lua')
| -rw-r--r-- | SpatialMatching.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/SpatialMatching.lua b/SpatialMatching.lua index 731d7ba..10e0dc7 100644 --- a/SpatialMatching.lua +++ b/SpatialMatching.lua @@ -14,10 +14,10 @@ end function SpatialMatching:updateOutput(input) -- input is a table of 2 inputs, each one being KxHxW - if self.full_output then + if self.full_output == 1 then self.output:resize(input[1]:size(2), input[1]:size(3), self.maxh, self.maxw) else - self.output:resize(input[1]:size(2)-self.maxh, input[1]:size(3)-self.maxw, + self.output:resize(input[1]:size(2)-self.maxh+1, input[1]:size(3)-self.maxw+1, self.maxh, self.maxw) end input[1].nn.SpatialMatching_updateOutput(self, input[1], input[2]) |