Fix bug in arguments of ImageTransformer, SPP and RCNN

fmassa · May 23, 2015 · 6e2d355 · 6e2d355
1 parent 17b12d7
commit 6e2d355
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/ImageTransformer.lua b/ImageTransformer.lua
@@ -5,7 +5,7 @@ function ImageTransformer:__init(...)
   self.raw_scale = 1
   self.swap = nil 
   local args = ...
-  for k,v in pairs(args) do self.k = v end
+  for k,v in pairs(args) do self[k] = v end
 end
 
 local function channel_swap(I,swap)

diff --git a/RCNN.lua b/RCNN.lua
@@ -11,7 +11,7 @@ function RCNN:__init(dataset,...)
   self.use_square = false
 
   local args = ...
-  for k,v in pairs(args) do self.k = v end
+  for k,v in pairs(args) do self[k] = v end
 end
 
 function RCNN:getCrop(im_idx,bbox,flip)

diff --git a/SPP.lua b/SPP.lua
@@ -26,7 +26,7 @@ function SPP:__init(dataset,model,...)
   self.cachedir = nil
 
   local args = ...
-  for k,v in pairs(args) do self.k = v end
+  for k,v in pairs(args) do self[k] = v end
 end