def print_classification_results(self,
test_data,
test_labels,
test_classes,
test_class_names=[],
normalize=True):
if self.transform is True:
test_data = data_transform.transform(
test_data, transform_method=self.transform_method)
feed_dict_test = {
self.x: test_data,
self.y_true: test_labels,
self.y_true_cls: test_classes
}
cls_true = test_classes
cls_pred = self.session.run(self.y_pred_cls, feed_dict=feed_dict_test)
plot_tools.plot_confusion_matrix(
cls_true,
cls_pred,
classes=test_class_names,
normalize=normalize,
title='Confusion matrix for test dataset using multi-layer '
'perceptron')
print('Detailed classification report')
print(
classification_report(y_true=cls_true,
y_pred=cls_pred,
target_names=test_class_names))
def predict(self, data):
if self.transform is True:
data = data_transform.transform(data, transform_method=self.transform_method)
feed_dict_data = {self.x: data}
predictions = self.session.run(self.y_pred_cls, feed_dict=feed_dict_data)
predictions = np.array(predictions)
return predictions
def load_test_data(self, data_directory='/tmp/cifar10/'):
"""
:param data_directory:
:return:
"""
print('Loading CIFAR 10 Test Dataset')
basic_dir_path = data_directory + 'cifar-10-batches/'
test_batch_path = 'test_batch'
test_files = [str(basic_dir_path + test_batch_path)]
print('Unpickling test file: %s' % test_files[0])
test_dict = [file_utils.unpickle(test_files[0])]
test_labels = []
test_images = []
print('Reading unpicked test file: %s' % test_files[0])
test_labels.extend(self.dict_read(test_dict[-1])[1])
test_images.extend(self.dict_read(test_dict[-1])[0])
test_images = np.array(test_images)
preprocessed_images = transform(test_images, transform_method=self.preprocess)
if self.make_image is True:
images = []
for fig_num in range(preprocessed_images.shape[0]):
fig = preprocessed_images[fig_num, :]
img = self.convert_images(fig, type=self.image_mode)
images.append(img)
images = np.array(images)
test_labels = np.array(test_labels)
self.test.data = np.array(preprocessed_images[:self.num_test_images])
if self.make_image is True:
self.test.images = np.array(images[:self.num_test_images, :])
self.test.fine_labels = np.array(test_labels[:self.num_test_images])
self.test.fine_class_names = np.array(list(map(lambda x: self.fine_classes[x], self.test.fine_labels)))
if self.one_hot_encode is True:
self.convert_one_hot_encoding(self.test.fine_labels, data_type='test')
if self.save_h5py != '':
h5f = h5py.File(self.save_h5py, 'a')
h5f.create_dataset('test_dataset', data=self.test.data, compression="gzip", compression_opts=9)
print('Written CIFAR 10 test dataset to file: %s' % self.save_h5py)
h5f.close()
del test_labels
del test_images
del preprocessed_images
if self.make_image is True:
del images
print()
return True
def load_test_data(self, data_directory='/tmp/cifar10/'):
print('Loading CIFAR 10 Test Dataset')
basic_dir_path = data_directory + 'cifar-10-batches/'
test_batch_path = 'test_batch'
test_files = [str(basic_dir_path + test_batch_path)]
print('Unpickling test file: %s' % test_files[0])
test_dict = [file_utils.unpickle(test_files[0])]
test_labels = []
test_images = []
print('Reading unpicked test file: %s' % test_files[0])
test_labels.extend(self.dict_read(test_dict[-1])[1])
test_images.extend(self.dict_read(test_dict[-1])[0])
test_images = np.array(test_images)
preprocessed_images = transform(test_images,
transform_method=self.preprocess)
if self.make_image is True:
images = []
for fig_num in range(preprocessed_images.shape[0]):
fig = preprocessed_images[fig_num, :]
img = self.convert_images(fig, type=self.image_mode)
images.append(img)
images = np.array(images)
test_labels = np.array(test_labels)
self.test.data = np.array(preprocessed_images[:self.num_test_images])
if self.make_image is True:
self.test.images = np.array(images[:self.num_test_images, :])
self.test.class_labels = np.array(test_labels[:self.num_test_images])
self.test.class_names = np.array(
list(map(lambda x: self.classes[x], self.test.class_labels)))
if self.one_hot_encode is True:
self.convert_one_hot_encoding(self.test.class_labels,
data_type='test')
del test_labels
del test_images
del preprocessed_images
if self.make_image is True:
del images
return True
def load_train_data(self, data_directory='/tmp/cifar100/'):
"""
:param data_directory:
:return:
"""
print('Loading CIFAR 100 Train Dataset')
basic_dir_path = data_directory + 'cifar-100-batches/'
data_batch_path = 'train'
data_files = [basic_dir_path + data_batch_path]
data_dict = [file_utils.unpickle(data_files[0])]
print('Reading unpicked data file: %s' % data_files[0])
data, fine_labels, coarse_labels, _, _ = self.dict_read(data_dict[0])
print(np.max(fine_labels))
print(np.max(coarse_labels))
data_fine_labels = fine_labels
data_coarse_labels = coarse_labels
data_images = np.array(data)
data_fine_labels = np.array(data_fine_labels)
data_coarse_labels = np.array(data_coarse_labels)
print('Success')
preprocessed_images = transform(data_images,
transform_method=self.preprocess)
if self.make_image is True:
images = []
for fig_num in range(preprocessed_images.shape[0]):
fig = preprocessed_images[fig_num, :]
img = self.convert_images(fig, type=self.image_mode)
images.append(img)
images = np.array(images)
if self.train_validate_split is None:
self.train.data = np.array(
preprocessed_images[:self.num_images, :])
if self.make_image is True:
self.train.images = np.array(images[:self.num_images, :])
self.train.fine_labels = np.array(
data_fine_labels[:self.num_images])
self.train.coarse_labels = np.array(
data_coarse_labels[:self.num_images])
self.train.fine_class_names = np.array(
list(
map(lambda x: self.fine_classes[x],
self.train.fine_labels)))
print(self.fine_classes[:15])
print(self.train.fine_labels[:15])
print(self.train.fine_class_names[:15])
self.train.coarse_class_names = np.array(
list(
map(lambda x: self.coarse_classes[x],
self.train.coarse_labels)))
else:
print('Requested to use only %d images' % self.num_images)
self.train.data = np.array(
preprocessed_images[:self.num_train_images, :])
if self.make_image is True:
self.train.images = np.array(images[:self.num_train_images, :])
self.train.fine_labels = np.array(
data_fine_labels[:self.num_train_images])
self.train.coarse_labels = np.array(
data_coarse_labels[:self.num_train_images])
self.train.fine_class_names = np.array(
list(
map(lambda x: self.fine_classes[x],
self.train.fine_labels)))
self.train.coarse_class_names = np.array(
list(
map(lambda x: self.coarse_classes[x],
self.train.coarse_labels)))
self.validate.data = \
np.array(preprocessed_images[self.num_train_images:self.num_train_images+self.num_validate_images, :])
if self.make_image is True:
self.validate.images = np.array(
images[self.num_train_images:self.num_train_images +
self.num_validate_images, :])
self.validate.fine_labels = \
np.array(data_fine_labels[self.num_train_images:self.num_train_images+self.num_validate_images])
self.validate.coarse_labels = \
np.array(data_coarse_labels[self.num_train_images:self.num_train_images + self.num_validate_images])
self.validate.fine_class_names = np.array(
list(
map(lambda x: self.fine_classes[x],
self.validate.fine_labels)))
self.validate.coarse_class_names = np.array(
list(
map(lambda x: self.coarse_classes[x],
self.validate.coarse_labels)))
if self.one_hot_encode is True:
self.convert_one_hot_encoding(self.train.fine_labels,
data_type='train',
class_type='fine')
self.convert_one_hot_encoding(self.train.coarse_labels,
data_type='train',
class_type='coarse')
if self.train_validate_split is not None:
self.convert_one_hot_encoding(self.validate.fine_labels,
data_type='validate',
class_type='fine')
self.convert_one_hot_encoding(self.validate.coarse_labels,
data_type='validate',
class_type='coarse')
if self.save_h5py != '':
h5f = h5py.File(self.save_h5py, 'a')
h5f.create_dataset('train_dataset',
data=self.train.data,
compression="gzip",
compression_opts=9)
print('Written CIFAR 100 train dataset to file: %s' %
self.save_h5py)
h5f.close()
del data_coarse_labels
del data_fine_labels
del data_images
del preprocessed_images
if self.make_image is True:
del images
print()
return True
def load_train_data(self, split=False, data_directory='/tmp/cifar10/'):
print('Loading CIFAR 10 Training Dataset')
basic_dir_path = data_directory + 'cifar-10-batches/'
data_batch_path = 'data_batch_'
data_files = []
data_dict = []
for i in range(1, 6):
data_files.append(str(basic_dir_path + data_batch_path + str(i)))
for file in data_files:
# print('Unpickling data file: %s' % file)
data_dict.append(file_utils.unpickle(file))
data_labels = []
data_images = []
for i in range(len(data_dict)):
print('Reading unpicked data file: %s' % data_files[i])
data, labels, _, _ = self.dict_read(data_dict[i])
data_labels.extend(labels)
data_images.extend(data)
data_images = np.array(data_images)
data_labels = np.array(data_labels)
preprocessed_images = transform(data_images,
transform_method=self.preprocess)
if self.make_image is True:
images = []
for fig_num in range(preprocessed_images.shape[0]):
fig = preprocessed_images[fig_num, :]
img = self.convert_images(fig, type=self.image_mode)
images.append(img)
images = np.array(images)
if self.train_validate_split is None:
self.train.data = np.array(
preprocessed_images[:self.num_images, :])
if self.make_image is True:
self.train.images = np.array(images[:self.num_images, :])
self.train.class_labels = np.array(data_labels[:self.num_images])
self.train.class_names = np.array(
list(map(lambda x: self.classes[x], self.train.class_labels)))
else:
print('Requested to use only %d images' % self.num_images)
self.train.data = np.array(
preprocessed_images[:self.num_train_images, :])
if self.make_image is True:
self.train.images = np.array(images[:self.num_train_images, :])
self.train.class_labels = np.array(
data_labels[:self.num_train_images])
self.train.class_names = np.array(
list(map(lambda x: self.classes[x], self.train.class_labels)))
self.validate.data = \
np.array(preprocessed_images[self.num_train_images:self.num_train_images+self.num_validate_images, :])
if self.make_image is True:
self.validate.images = np.array(
images[self.num_train_images:self.num_train_images +
self.num_validate_images, :])
self.validate.class_labels = \
np.array(data_labels[self.num_train_images:self.num_train_images+self.num_validate_images])
self.validate.class_names = np.array(
list(map(lambda x: self.classes[x],
self.validate.class_labels)))
if self.one_hot_encode is True:
self.convert_one_hot_encoding(self.train.class_labels,
data_type='train')
if self.train_validate_split is not None:
self.convert_one_hot_encoding(self.validate.class_labels,
data_type='validate')
del data_labels
del data_images
del preprocessed_images
if self.make_image is True:
del images
return True
def optimize(self,
train_data,
train_labels,
train_classes,
validate_data=None,
validate_labels=None,
validate_classes=None,
test_data=None,
test_labels=None,
test_classes=None):
if self.transform is True:
train_data = data_transform.transform(
train_data, transform_method=self.transform_method)
if self.train_validate_split is not None:
validate_data = data_transform.transform(
validate_data, transform_method=self.transform_method)
if self.test_log is True:
test_data = data_transform.transform(
test_data, transform_method=self.transform_method)
file_name = os.path.splitext(os.path.abspath(self.model_name))[0]
num_files = len(
sorted(glob.glob(os.path.abspath(file_name + '*.meta'))))
if num_files > 0:
checkpoint_file = os.path.abspath(
sorted(glob.glob(file_name + '*.data-00000-of-00001'),
reverse=True)[0])
if os.path.exists(checkpoint_file):
print('Restoring model from %s' % checkpoint_file)
meta_file = os.path.abspath(
sorted(glob.glob(file_name + '*.meta'), reverse=True)[0])
print('Loading: %s' % meta_file)
saver = tf.train.import_meta_graph(meta_file)
print('Loading: %s' % os.path.abspath(checkpoint_file))
cpk = tf.train.latest_checkpoint(os.path.dirname(meta_file))
print('Checkpoint: ' + str(cpk))
print('Tensors')
print(
print_tensors_in_checkpoint_file(file_name=cpk,
all_tensors='',
tensor_name=''))
saver.restore(
self.session,
tf.train.latest_checkpoint(os.path.dirname(meta_file)))
print('Last epoch to restore: ' +
str(self.session.run(self.global_step)))
if self.train_validate_split is not None:
if self.test_log is False:
self.run(train_data,
train_labels,
train_classes,
validate_data=validate_data,
validate_labels=validate_labels,
validate_classes=validate_classes)
else:
self.run(train_data,
train_labels,
train_classes,
validate_data=validate_data,
validate_labels=validate_labels,
validate_classes=validate_classes,
test_data=test_data,
test_labels=test_labels,
test_classes=test_classes)
else:
if self.test_log is False:
self.run(train_data, train_labels, train_classes)
else:
self.run(train_data,
train_labels,
train_classes,
test_data=test_data,
test_labels=test_labels,
test_classes=test_classes)
