Как читать из файлов с расширением tfrec в tensorflow

#python #tensorflow #tensorflow2.0

#python #tensorflow #tensorflow2.0

Вопрос:

Я новичок в TensorFlow, и мне трудно понять, как читать файлы с расширением tfrec, в которых хранятся изображения, я прочитал документацию, но она очень запутанная, можете ли вы помочь мне решить эту проблему?

Комментарии:

1. Можете ли вы добавить пример кода, который вы пробовали?

Ответ №1:

 files = tf.data.Dataset.list_files(filenames)
dataset = files.apply(tf.contrib.data.parallel_interleave(tf.data.TFRecordDataset,
                                                          cycle_length=argument.num_parallel_readers, sloppy=True))
dataset = dataset.map(input_parse_exmp, num_parallel_calls=16)
# Shuffle, repeat, and batch the examples.
dataset = dataset.batch(argument.batch_size)