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

Question

我有几个 tfrecord 分成 5 个分片

例如） train-1.tfrecord train-2.tfrecord train-3.tfrecord train-4.tfrecord train-5.tfrecord

我想使用所有的 tfrecord。

我的检测模型是 Faster-rcnn(resnet101)

例如） 模型/研究/object_detection/samples/configs/faster_rcnn_resnet101.config

我可以这样使用吗？

.. input_path: "/path/to/train-*.tfrecord" ..

Answer 1

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

filenames = tf.gfile.Glob(input_files)

所以您的示例应该按预期工作。