python - 在 TensorFlow 中读取 jpeg 图像列表并将其转换为 TFRecords 的最快方法是什么？

Question

我有一个 txt 文件中的图像和标签列表，格式如下：

文件1.jpg 1

文件2.jpg 0

文件3.jpg 1

文件4.jpg 1

其中标签是二进制的。

如何将图像数据转换为 TFRecord 文件而不将我的图像分配给标签的子目录（用于初始模型数据准备），并且只需读取 jpg 图像？我已经看过 tensorflow 的 TFRecords 教程，但似乎图像已经全部以张量形式构建。

我尝试了一个解决方案： 如何将 jpeg 图像目录转换为 tensorflow 中的 TFRecords 文件？

但它似乎没有用。

Answer 1

按照以下链接中的代码解决您的问题。我完全按照你的要求做了。（将一组 png 或 jpeg 图像转换为 TfRecords（二进制图像）。我提供了读取 tfrecords 的方法。此外，就我而言，我有一组存储在字符串中的特征，并且我将它们存储到 tfrecords 中。

从字符串列表创建 TfRecords 并在解码后在 tensorflow 中输入 Graph

希望这段代码对您有所帮助。