mnist

mnist.py

import tensorflow as tf
import matplotlib.pyplot as plt

from datasets import mnist

dataset = mnist.train()
iterator = dataset.make_one_shot_iterator()
next_image, next_label = iterator.get_next()

with tf.Session() as sess:
    next_image, next_label = sess.run([next_image, next_label])
    next_image = np.reshape(next_image, [28, 28])
    print(next_label)
    plt.imshow(next_image)
    plt.show()