python - 如何从张量流中具有 tfrec 扩展名的文件中读取

Question

我是 TensorFlow 的初学者，我很难理解如何读取存储图像的 tfrec 扩展文件，我阅读了文档但它非常混乱，你能帮我解决这个问题吗？

标签： pythontensorflowtensorflow2.0

Answer 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)