szagoruyko · October 25, 2016 10:08
diff --git a/cifar100_make_original.lua b/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)
	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)