在Tensorflow中准备我自己的数据集

时间:2016-10-26 03:49:27

标签: python tensorflow

您好我想知道如何在Tensorflow中准备我自己的数据集 我有.PNG格式的数千张图像,标签存储在.TXT格式的文本文件中(每个标签用新行分隔)。我如何准备他们的tensorflow训练? PNG文件名称类似于1.png,2.png,3.png等等... TXT文件包含由新行分隔的标签(每行包含一个标签) 如果有人能给出一些具体的例子,我们将非常感激。

1 个答案:

答案 0 :(得分:-1)

通常在tensorflow中你编写一个读取输入的小python程序, 无论采用何种格式,都可以转换为int或float数组并将其转换为 通过占位符的张量流图。

MNIST教程中有一个例子: https://www.tensorflow.org/versions/r0.11/tutorials/mnist/beginners/index.html

他们阅读数据的方式是

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

我从未使用过图像,所以我不知道是否使用了一些标准文件 格式以及是否有工具以该格式生成数据。 希望这会有所帮助。