Imported DataSetLabelMe, unmodified.

1 files changed, 11 insertions, 5 deletions

diff --git a/DataSet.lua b/DataSet.lua
index 81e4bf3..23faf73 100644
--- a/DataSet.lua
+++ b/DataSet.lua
@@ -1,7 +1,7 @@
 --------------------------------------------------------------------------------
 -- DataSet: a class to handle standard datasets.
 --
--- Authors: Corda, Farabet
+-- Authors: Clement Farabet, Benoit Corda
 --------------------------------------------------------------------------------
 
 local lDataSet = torch.class('nn.DataSet')
@@ -18,6 +18,16 @@ function lDataSet:size()
    return self.nbSamples
 end
 
+function lDataSet:__tostring__()
+   str = 'DataSet:\n'
+   if self.nbSamples then
+      str = str .. ' + nb samples : '..self.nbSamples
+   else
+      str = str .. ' + empty set...'
+   end
+   return str
+end
+
 function lDataSet:load(...)
    -- parse args
    local args, dataSetFolder, nbSamplesRequired, cacheFile, channels, sampleSize
@@ -304,10 +314,6 @@ function lDataSet:display(args) -- opt args : scale, nbSamples
    end
 end
 
-function lDataSet:__show()
-   self:display{nbSamples=100}
-end
-
 function lDataSet:useCacheFile(fileName)
    self.cacheFileName = fileName
 end