Tensorflow对象检测API TFRecord

Question

我是tensorflow TFRecord的新手。所以我正在研究Tensorflow对象检测API代码

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/using_your_own_dataset.md

但是我找不到加载tfrecord的代码。 我认为他们使用.config文件加载tfrecord，因为我是在配置文件中找到的。

tf_record_input_reader {  
input_path: "/path/to/train_dataset.record-?????-of-00010" 
}

有人可以帮忙吗？

Answer 1

您是否已将数据集转换为TFRecord格式？ 如果是这样，您应该有一个路径，其中包含将您的训练数据集分片到几个记录文件中，格式为 <path_to_training_data>/<train_dataset>.record-?????-of-xxxxx

其中<path_to_training_data>是上述训练数据集的路径，<train_dataset>是您赋予每个文件的文件名，xxxxx是创建的记录文件数（例如00010），和?????应该保留不变，并用作所有记录文件的格式。 将<path_to_training_data>，<train_dataset>和xxxxx替换为数据集的正确值之后，TF OD API应该处理其他所有事情（查找所有记录，读取它们，等等）。 / p>

请注意，训练数据集和评估数据集通常都使用tf_record_input_reader（验证/测试），并且每个都应具有相应的上述值（路径，数据集名称，文件数）。