Где изображения читаются в этой программе Python классификации?

Question

Я смотрю на этот питон код доктора Мартина, где он делает рукописную идентификацию номера.Я пытаюсь понять, где я могу вставить свои собственные изображения и метки для классификации.Я не мог найти, где он читает номера изображений и меток.

Этот код находится в папке Github .

Answer 1

Используется предопределенный набор данных MNIST:

import mnistdata

Данные считываются в строке 38:

mnist = mnistdata.read_data_sets("data", one_hot=True, reshape=False)

И передаются в сеть здесь:

    batch_X, batch_Y = mnist.train.next_batch(100)

    # compute training values for visualisation
    if update_train_data:
        a, c, im, w, b, l = sess.run([accuracy, cross_entropy, I, allweights, allbiases, lr],
                                 feed_dict={X: batch_X, Y_: batch_Y, step: i})