替换 TensorFlow CNN 模型中的 MNIST 数据答案

【问题标题】：Replace MNIST data in Tensorflow CNN models替换 TensorFlow CNN 模型中的 MNIST 数据
【发布时间】：2017-12-28 12:04:03
【问题描述】：

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

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

【问题讨论】：

标签： python tensorflow

【解决方案1】：

您可能会查看this 文件，尤其是在对 mnist 数据的导入进行编程的行处。

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

此函数导入 mnist 数据。根据the function's sourcecode（第 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'

或者，您可以根据需要调整函数的源代码并编写自己的导入函数。

【讨论】：

好的，我重写了我的函数以便具有相同的名称。但我仍然不知道如何读取我的数据。在 Tensorflow 教程中，数据是自动加载的（就像在您链接的 mnist_softmax.py 代码中一样），所以我不知道如何阅读这些 IDX1-UBYTE 文件。
当您只想像教程函数中那样读取训练文件时，您可以使用 mnist_softmax.py 第 65 行和第 71/72 行中使用的数据。如果你想用一个你想从头开始编写的函数来读取这些文件，我不得不承认，我无法帮助你。