generate mnist images using keras

create_mnist_with_keras.py

from keras.datasets import mnist
from PIL import Image

(x_train, y_train), (x_validate, y_validate) = mnist.load_data()

dataset_train = {'inputs': x_train, 'labels': y_train}
dataset_validate = {'inputs': x_validate, 'labels': y_validate}

for index, i in enumerate(dataset_train['inputs']):
    im = Image.fromarray(i)
    file_name = "{}_{}.jpg".format(dataset_train['labels'][index], index)
    directory = os.path.join(DATASET_DIR, "mnist/train/{}".format(dataset_train['labels'][index]))
    im.save(os.path.join(directory, file_name))
    
for index, i in enumerate(dataset_validate['inputs']):
    im = Image.fromarray(i)
    file_name = "{}_{}.jpg".format(dataset_validate['labels'][index], index)
    directory = os.path.join(DATASET_DIR, "mnist/validate/{}".format(dataset_validate['labels'][index]))
    im.save(os.path.join(directory, file_name))