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
]
Это правильный способ прочитать все записи одновременно? И если да, что мне нужно изменить, чтобы на самом деле читать записи? Спасибо много!