def load_test_data(self, data_dim=2, one_hot=True, norm=True): print('Extract test data: [mnist] from {}'.format(self.data_path)) data = sio.loadmat('{}/mnist_test.mat'.format(self.data_path)) images = data['images'].astype(np.float32) labels = data['labels'].astype(np.float32) images, labels = self.preprocess(images, labels, data_dim=data_dim, one_hot=one_hot, norm=norm) # package into iterator assert self.package == None self.package = iter.Iterator(images, labels)
def load_train_data(self, data_dim=4, one_hot=True, value_norm=True): print('Extract train data: [cifar10] from {}'.format(self.data_path)) data = sio.loadmat('{}/cifar10_train.mat'.format(self.data_path)) images = data['images'].astype(np.float32) labels = data['labels'].astype(np.float32) images, labels = self.preprocess(images, labels, data_dim=data_dim, one_hot=one_hot, value_norm=value_norm) # package into iterator assert self.package == None self.package = iter.Iterator(images, labels)