Search code examples
tensorflowobject-detection-apitfrecord

Set multiple tfrecord into config file of Tensorflow Object Detection API

I have several tfrecord splited into 5 shards

ex) train-1.tfrecord train-2.tfrecord train-3.tfrecord train-4.tfrecord train-5.tfrecord

And I want to use all tfrecord.

My detection model is Faster-rcnn(resnet101)

ex) models/research/object_detection/samples/configs/faster_rcnn_resnet101.config

can I use like this?

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

Solution

  • Yes, you can use 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)

    So your example should work as expected.