cnn 使用 tensorflow 处理自己的图像集 - tfrecord 格式应该是什么答案

【问题标题】：cnn using tensorflow for own image set - what should be the tfrecord formatcnn 使用 tensorflow 处理自己的图像集 - tfrecord 格式应该是什么
【发布时间】：2016-12-23 19:27:28
【问题描述】：

我有一个大小为 600 x 400 的图像数据集，并且我已将每个图像转换为 TFRecord 格式。但我无法弄清楚如何使用这些数据？我已经看到了 imagenet 数据集，发现只有一个二进制文件（从 here 提取时）。

是图像数据集只有一个 TFRecord 还是每个图像都有自己的 TFRecord 文件？

【问题讨论】：

标签： tensorflow neural-network

【解决方案1】：

Tensorflow 不会寻找单个 tfrecord 文件。因此，请随意将您的“数据目录”和“训练目录”指向包含 tfrecord 文件集的位置。

另外，请记住文件应根据其名称位于相应的目录中，例如“train directory”中的 TRAIN-*.tfrecord 文件。

如果您提到要在这些 TF 记录文件上运行的 TF 模型，答案可能会更具体。

希望对你有帮助。

【讨论】：

您能解释一下如何加载我拥有的数据集，即在我的train_directory 中我有 5 个名称为 label_i 的标签文件夹（i={1,2,3,4,5 })。
^ 在每个 label_i 文件夹中，我都有单独的图像 TF Record 文件。
错了！不要为每个“标签”/“类”文件夹创建单独的 TFRecord 文件。在生成 TFrecord 文件时，指向其中包含标签目录的父 data_dir，并在 build_dataset 脚本中指定您想要在训练中使用多少图像以及在评估中需要多少图像