首页 文章

替换Tensorflow CNN模型中的MNIST数据

提问于
浏览 次
0

我试图在张量流中训练我自己的数据集上的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 回答

  • 0

    您可以查看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行），您的文件应具有以下名称，因此您需要重写已链接的函数，以便有一个测试集 .

    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'

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

    回复于

相关问题