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

Set multiple tfrecord into config file of Tensorflow Object Detection API

我有几个 tfrecord 分成 5 个碎片

例如) 火车-1.tfrecord 火车-2.tfrecord 火车-3.tfrecord 火车-4.tfrecord 火车-5.tfrecord

我想使用所有的 tfrecord。

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

例如) models/research/object_detection/samples/configs/faster_rcnn_resnet101.config

我可以这样使用吗?

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

是的,您可以使用 glob patterns in the file path that you give (or paths, since you can give more than one too). Although I couldn't find explicit documentation about it, looking at the source code, you can see that the input paths are expanded with tf.gfile.Glob:

filenames = tf.gfile.Glob(input_files)

因此您的示例应该按预期工作。