cifar100_make_original.lua

require 'xlua'
require 'sys'
local matio = require 'matio'

local data = {}
local labels = {}


local train_data = matio.load'./cifar-100-matlab/train.mat'
local test_data = matio.load'./cifar-100-matlab/test.mat'

local n_train = train_data.fine_labels:numel()
local n_test = test_data.fine_labels:numel()

local dataset = {
   trainData = {
      data = train_data.data:clone():view(-1,3,32,32),
      labels = train_data.fine_labels:squeeze() + 1,
      size = function() return n_train end,
   },
   testData = {
      data = test_data.data:clone():view(-1,3,32,32),
      labels = test_data.fine_labels:squeeze() + 1,
      size = function() return n_test end,
   }
}

print(dataset)
print(dataset.trainData.labels:max())
print(dataset.testData.labels:max())
torch.save('cifar100_original.t7', dataset)