小编Mar*_*ach的帖子

从TFRecordDataset获取数据集为numpy数组

我正在使用新的tf.dataAPI为CIFAR10数据集创建迭代器。我正在从两个.tfrecord文件读取数据。一个保存训练数据(train.tfrecords),另一个保存测试数据(test.tfrecords)。一切正常。但是,在某些时候,我需要两个数据集(训练数据和测试数据)都为numpy数组

是否有可能从tf.data.TFRecordDataset对象检索作为numpy数组的数据集?

python numpy tensorflow tensorflow-datasets

5
推荐指数
1
解决办法
2124
查看次数

标签 统计

numpy ×1

python ×1

tensorflow ×1

tensorflow-datasets ×1