首页 > 解决方案 > 如何从多个 TFRecord 文件中的数据创建 TensorFlow 2 生成器?

问题描述

我想创建一个从多个 TFRecord 生成数据的自定义生成器。

标签: tensorflowkerasdeep-learningtfrecord

解决方案


指定 a 时可以指定多个文件tf.data.TFRecordDataset

dataset = tf.data.TFRecordDataset(["record1.tfrecords", "record2.tfrecords"])


推荐阅读