Numpy数组到TFrecord

Question

我正在尝试通过张量流对象检测API训练自定义数据集。数据集包含numpy ndarray格式（uint8）的40k训练图像和标签。训练数据集shape = 2（[40000,23456]）且标签shape = 1（[0 ...，3]）。我想为此数据集生成tfrecord。我怎么做？我不喜欢tensorflow。

Answer 1

本教程将引导您完成从您的数据创建 TFRecords 的过程：

https://medium.com/mostly-ai/tensorflow-records-what-they-are-and-how-to-use-them-c46bc4bbb564

但是，现在使用 Dataset 输入管道有更简单的预处理方法。我更喜欢将我的数据保留为最原始的格式，并构建一个预处理管道来处理它。这是您想要阅读以了解数据集预处理管道的主要指南：

https://www.tensorflow.org/programmers_guide/datasets