将多个tfrecord设置到Tensorflow对象检测API的配置文件中

Question

我有几个tfrecord分成5个碎片

ex）train-1.tfrecordtrain-2.tfrecordtrain-3.tfrecord火车4.tfrecordtrain-5.tfrecord

而且我想使用所有tfrecord。

我的检测模型是Faster-rcnn（resnet101）

ex）模型/研究/ object_detection /样本/配置/faster_rcnn_resnet101.config

我可以这样使用吗？

..input_path：“ / path / to / train-*。tfrecord”..

Answer 1

是，您可以在您提供的文件路径中使用glob patterns（或路径，因为您也可以提供多个）。尽管我找不到关于它的明确文档，但查看source code，您可以看到输入路径已使用tf.gfile.Glob展开：

tf.gfile.Glob

所以您的示例应该可以正常工作。