我如何在Tensorflow中准备自己的数据集(数千张图片)?

时间:2016-10-24 06:50:11

标签: python tensorflow

我有几千张.PNG格式的图片。这些png文件的标签存储在txt文件中。 PNG文件名称类似于1.png,2.png,3.png等等... TXT文件包含由新行分隔的标签(每行包含一个标签)

如何准备它们用于tensorflow?

准备好/加载后,我想通过使用上面提到的图像来训练张量流中的CNN。 如果有人能够提供在tensorflow中读取png和txt文件的具体示例,将非常感激。

1 个答案:

答案 0 :(得分:1)

您可能想查看tensorflow uint32_t a = 0; char c; while ((c = getchar()) != '\n') { // read a line char by char a <<= 1; // shift the uint32 a bit left a += (c - '0') & 1; // convert the char to 0/1 and put it at the end of the binary } printf("%u\n", a); 库,包括其slim类和dataset的使用

苗条:https://github.com/tensorflow/tensorflow/tree/r0.11/tensorflow/contrib/slim

数据集: https://github.com/tensorflow/tensorflow/tree/r0.11/tensorflow/contrib/slim/python/slim/data