TensorFlow - читать все примеры из TFRecords сразу?

Как вы читаете все примеры из TFRecords одновременно?

Я используюtf.parse_single_example читать отдельные примеры, используя код, аналогичный приведенному в методеread_and_decode впример полностью_коннектированного читателя, Однако я хочу запустить сеть по всему набору данных для проверки сразу, и поэтому хотел бы загрузить их полностью.

Я не совсем уверен, нодокументация кажется, я могу использоватьtf.parse_example вместоtf.parse_single_example загрузить весь файл TFRecords сразу. Я не могу заставить это работать, хотя. Я предполагаю, что это связано с тем, как я определяю функции, но я не уверен, как в спецификации функции утверждать, что есть несколько примеров.

Другими словами, моя попытка использовать что-то похожее на:

reader = tf.TFRecordReader()
_, serialized_example = reader.read(filename_queue)
features = tf.parse_example(serialized_example, features={
    'image_raw': tf.FixedLenFeature([], tf.string),
    'label': tf.FixedLenFeature([], tf.int64),
})

не работает, и я предполагаю, что это потому, что функции не ожидают нескольких примеров одновременно (но опять же, я не уверен). [Это приводит к ошибкеValueError: Shape () must have rank 1]

Это правильный способ прочитать все записи одновременно? И если да, что мне нужно изменить, чтобы на самом деле читать записи? Спасибо много!

Ответы на вопрос(6)

Ваш ответ на вопрос