Numpy 数组到 TFrecord
Numpy array to TFrecord
我正在尝试通过 tensorflow 对象检测来训练自定义数据集 api。数据集包含 40k 训练图像和标签，这些图像和标签采用 numpy ndarray 格式 (uint8)。训练数据集 shape=2 ([40000,23456]) 和标签 shape = 1 ([0..., 3])。我想为此数据集生成 tfrecord。我该怎么做？
本教程将引导您完成从数据创建 TFRecords 的过程：
https://medium.com/mostly-ai/tensorflow-records-what-they-are-and-how-to-use-them-c46bc4bbb564
但是，现在使用数据集输入管道可以更轻松地处理预处理。我更喜欢以最原始的格式保存我的数据，并构建一个预处理管道来处理它。以下是您想要阅读以了解数据集预处理管道的主要指南：
我正在尝试通过 tensorflow 对象检测来训练自定义数据集 api。数据集包含 40k 训练图像和标签，这些图像和标签采用 numpy ndarray 格式 (uint8)。训练数据集 shape=2 ([40000,23456]) 和标签 shape = 1 ([0..., 3])。我想为此数据集生成 tfrecord。我该怎么做？
本教程将引导您完成从数据创建 TFRecords 的过程：
https://medium.com/mostly-ai/tensorflow-records-what-they-are-and-how-to-use-them-c46bc4bbb564
但是，现在使用数据集输入管道可以更轻松地处理预处理。我更喜欢以最原始的格式保存我的数据，并构建一个预处理管道来处理它。以下是您想要阅读以了解数据集预处理管道的主要指南：