使用TFRecord训练/测试文件

Question

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

在阅读Tensorflow相关帖子后，我意识到TFRecord是最合适的文件格式。因此，我正在寻找使用TFRecord数据的完整CNN示例。我注意到图像相关的教程（link1和link2中的mnist和cifar10）提供了不同的二进制文件格式，其中整个数据集一次加载。因此，我想知道是否有人知道这些教程（mnist和cifar10）是否可以使用TFRecord数据（对于CPU和GPU）。

Answer 1

我假设您要同时写入和读取TFRecord文件。我认为这里所做的reading_data.py应该可以帮助您将MNIST数据转换为TFRecors。 为了阅读它，这个脚本可以解决问题：fully_connected_reader.py 这可以与cifar10类似地完成。