具有TFRecord列车/测试文件的mnist和cifar10示例

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来完成。