Установить несколько tfrecord в файл конфигурации API обнаружения объектов Tensorflow - PullRequest
1 голос
/ 25 октября 2019

У меня есть несколько записей, разделенных на 5 осколков

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

И я хочу использовать все tfrecord.

Моя модель обнаружения - Faster-rcnn (resnet101)

ex) модели / исследования / object_detection / samples / configs / fast_rcnn_resnet101.config

Могу ли я использовать вот так?

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

1 Ответ

0 голосов
/ 25 октября 2019

Да, вы можете использовать шаблоны glob в указанном вами пути к файлу (или путях, поскольку вы можете указать и более одного). Хотя я не смог найти точную документацию об этом, глядя на исходный код , вы можете увидеть, что входные пути расширены с помощью tf.gfile.Glob:

filenames = tf.gfile.Glob(input_files)

Так что ваш пример должен работать как положено.

...