diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2011-08-16 20:28:48 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2011-08-16 20:28:48 +0400 |
commit | bc116e653147dfe944dab37f07113e138b11d55b (patch) | |
tree | acad1381a15ec5f4897c6eec4d0290ca2418420d | |
parent | cd55b008834743dc25f54f9b3c82e30f8b9d832a (diff) |
Fixed missing index (matlab mode for DataSetLabelMe)
-rw-r--r-- | DataSetLabelMe.lua | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/DataSetLabelMe.lua b/DataSetLabelMe.lua index a416325..decec18 100644 --- a/DataSetLabelMe.lua +++ b/DataSetLabelMe.lua @@ -297,6 +297,8 @@ function DataSetLabelMe:loadSample(index) mask_loaded = matrix break end + img_loaded = img_loaded:transpose(2,3) + mask_loaded = mask_loaded:transpose(1,2) else img_loaded = image.load(self.rawdata[index].imgfile) mask_loaded = image.load(self.rawdata[index].maskfile)[1] @@ -337,8 +339,8 @@ function DataSetLabelMe:loadSample(index) end else self.currentMask:mul(self.nbClasses-1):add(0.5):floor():add(1) - self.currentIndex = index end + self.currentIndex = index end end |