在Tensorflow CNN模型中替换MNIST数据

Question

我正在尝试在张量流中的自己的数据集上训练CNN模型。 我使用以下代码https://github.com/gskielian/JPG-PNG-to-MNIST-NN-Format/blob/master/convert-images-to-mnist-format.py将数据转换为相同的MNIST格式。

现在我不知道如何用我自己的数据替换MNIST数据。 有没有办法用我自己的数据替换mnist数据或将它们导入tensorflow？

Answer 1

您可能会看一下此文件，特别是在mnist数据导入的编程行中。

from tensorflow.examples.tutorials.mnist import input_data    
mnist = input_data.read_data_sets(FLAGS.data_dir, one_hot=True)

此函数导入mnist数据。 根据函数的源代码 （229 ff行），您的文件应具有以下名称，因此您需要重写链接到的函数，以提供一个测试集。

TRAIN_IMAGES = 'train-images-idx3-ubyte.gz'
TRAIN_LABELS = 'train-labels-idx1-ubyte.gz'
TEST_IMAGES = 't10k-images-idx3-ubyte.gz'
TEST_LABELS = 't10k-labels-idx1-ubyte.gz'

另外，您可以根据需要调整函数的来源，并编写自己的导入函数。