In [7]:
import matplotlib.pyplot as plt
import numpy as np
from iam_dataset import IamLinesDataset
dataset = IamLinesDataset()
dataset.load_or_generate_data()


BIG DOG (10000, 10, 28)
BIG DOG Test (1000, 10, 28)

In [2]:
dataset.x_train.shape


Out[2]:
(10000, 28, 280)

In [11]:
plt.figure(figsize = (25, 1))
offset = 9999
plt.imshow(dataset.x_train[offset], cmap='gray', interpolation='none')


Out[11]:
<matplotlib.image.AxesImage at 0x7ff39ae27fd0>

In [3]:
dataset.y_train.shape


Out[3]:
(7101, 97, 80)

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])


[18  4 13  8 14 17 27 27 27 27]
Out[12]:
'senior____'

In [ ]: