mnist手写数字集制作tfrecords数据格式

一、tfrecords文件

tfrecords是一种二进制文件，可先将图片与标签制作成该格式的文件，使用tfrecords进行数据读取，会提高内存利用率，将不同输入文件统一起来。

二、mnist数据集

MNIST数据集是一个手写体数字集合，可到此处下载，数据包括四部分：训练图片集、训练标签集、测试图片集、测试标签集。该数据集的训练集中有55000张图片，验证集中5000有张图片，测试集中是10000张图片。

三、数据制作与读取

1、生成文件

文件生成的过程：

• 新建一个writer
• for循环遍历每张图和标签
• 把每张图和标签封装到example中
• 将example序列化

具体代码如下：

def generate_tfRecord():
    isExists=os.path.exists(data_path)   ##判断保存路径是否存在
    if not isExists:
        os.makedirs(data_path)
        print('路径创建成功')
    else:
        print('路径已存在')
    write_tfR