In [7]:
import matplotlib.pyplot as plt
import numpy as np
from iam_dataset import IamLinesDataset
dataset = IamLinesDataset()
dataset.load_or_generate_data()
In [2]:
dataset.x_train.shape
Out[2]:
In [11]:
plt.figure(figsize = (25, 1))
offset = 9999
plt.imshow(dataset.x_train[offset], cmap='gray', interpolation='none')
Out[11]:
In [3]:
dataset.y_train.shape
Out[3]:
In [12]:
def convert_y_label_to_string(y, dataset=dataset):
return ''.join([dataset.mapping[i] for i in np.argmax(y, axis=-1)])
print(np.argmax(dataset.y_train[offset], axis=-1))
convert_y_label_to_string(dataset.y_train[offset])
Out[12]:
In [ ]: