CODEWITHSUNDEEP
tensorflowobject-detection-apitfrecord
yKim
yKim

Reputation: 63

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" ..

Upvotes: 2

Views: 635

Answers (1)

javidcf
javidcf

Reputation: 59701

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.

Upvotes: 1

Related Questions