该分类python程序在哪里读取图像？

Question

我正在看Martin博士的这个Python code，它可以手写识别数字。我试图了解在哪里可以插入自己的图像和标签进行分类。我找不到它读取数字图像和标签的位置。

此代码位于Github folder。

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