python - 如何将 TFRecord 数据集映射到 pbtxt 文件中的标签？

Question

一般来说，我对 Tensorflow 和 ML 非常缺乏经验。我正在尝试遵循https://www.tensorflow.org/tutorials/images/transfer_learning上的教程，但我的数据集的结构如下：我有一个train.tfrecord文件和一个train_labels.pbtxt带有标签的文件，这些标签告诉每个图像中不同对象的位置位于（验证和测试集也是如此）。所以我需要train_dataset在运行以下代码之前将它们映射到：

history = model.fit(train_dataset,
                    epochs=initial_epochs,
                    validation_data=validation_dataset)

我知道在将数据集与keras.applications所选模型一起使用之前，我必须对其进行规范化，而且我可能必须使用tf.data.TFRecordDataset()，但我根本不理解文档。我看过一些教程，他们使用管道配置文件，它们只是指示这些文件的路径，但这不是我想要做的。

谢谢您的帮助。

标签： pythontensorflowdatasettfrecord