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

Question

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

file1.jpg 1

file2.jpg 0

file3.jpg 1

file4.jpg 1

标签是二进制的。

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

我尝试过以下解决方案： How do I convert a directory of jpeg images to TFRecords file in tensorflow?

但它似乎没有效果。

Answer 1

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

Creating TfRecords from a list of strings and feeding a Graph in tensorflow after decoding

希望此代码能为您提供帮助。