带有TFRecord训练/测试文件的mnist和cifar10示例

Question

我是Tensorflow的新用户。我想用它来训练一个有2M张图像的数据集。我在caffe中使用lmdb文件格式做了这个实验。

在阅读了Tensorflow的相关文章后，我意识到TFRecord是最适合这样做的文件格式。因此，我正在寻找使用TFRecord数据的完整CNN示例。我注意到与图像相关的教程( link1和link2中的mnist和cifar10 )提供了一种不同的二进制文件格式，其中一次性加载整个数据集。因此，我想知道是否有人知道这些教程(mnist和cifar10)是否可以使用TFRecord数据(用于中央处理器和图形处理器)。

Answer 1

我假设您想同时写入和读取TFRecord文件。我认为这里所做的reading_data.py应该可以帮助您将MNIST数据转换为TFRecors。为了回读它，这个脚本做了这个把戏：fully_connected_reader.py这可以用cifar10类似的方式来完成。