سلام . به نظرم دیتاست را از سایتش دانلود کنید خیلی بهتره.
دریافت دیتاست از طریق tensrflow به صورت زیر :
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
train_X,train_Y = mnist.train.images,mnist.train.labels
test_X,test_Y = mnist.test.images,mnist.test.labels
val_X,vali_Y = mnist.validation.images,mnist.validation.labels