Разбор файлов, использованных при создании eval.tfrecord - PullRequest
1 голос
/ 11 июля 2019

Я пытаюсь получить файлы .jpg, используемые для создания eval.tfrecord.

У меня проблемы с открытием файла с помощью python и выполнением регулярного выражения в файле, чтобы найти все файлы .jpg.перечислены случайным образом по всему файлу.

Теперь у меня есть файл, открытый с использованием приведенного ниже кода и вижу один из файлов .jpg мне нужно в структурированном виде:

  feature {
    key: "image/source_id"
    value {
      bytes_list {
        value: "path/to/image.jpg"
      }
    }
  }

Вот кодчтобы получить приведенный выше результат:

#From the Tensorflow Docs:

from __future__ import absolute_import, division, print_function, unicode_literals

import tensorflow as tf
tf.enable_eager_execution()

import numpy as np
import IPython.display as display

record_iterator = tf.python_io.tf_record_iterator("eval.tfrecord")

for string_record in record_iterator:
  example = tf.train.Example()
  example.ParseFromString(string_record)

  print(example)

  # Exit after 1 iteration as this is purely demonstrative.
  break

Я хотел бы просто использовать .jpgs, а не все данные в файле.Может быть, есть какой-то способ указать ключ: image / source_id, чтобы получить все файлы .jpgs, использованные при создании tfrecord?

1 Ответ

0 голосов
/ 12 июля 2019

Я перенаправил вывод «print (example)» в файл.Затем смог прочитать этот файл и разобрать его как обычно, используя регулярные выражения для поиска файлов .jpg.

...