tensorflow实战之TFRecord数据的写入与读取

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
import matplotlib.pyplot as plt

#生成整数型属性
def int64List(values):
    return tf.train.Feature(int64_list=tf.train.Int64List(value=[values]))
#生成字符串型属性
def bytesList(values):
    return tf.train.Feature(bytes_list=tf.train.BytesList(value=[values]))
#定义一个类用来写入
writer = tf.python_io.TFRecordWriter('path/mnist.tfrecord')
#加载数据
mnist = input_data.read_data_sets('data/')
#写入
num = mnist.train.num_examples
images = mnist.train.images
labels = mnist.train.labels
for index in range(num):
    images_raw = images[index].tostring() 
    label = labels[index]
    example = tf.train.Example(features=tf.train.Features(feature={
        'labels':int64List(label),
        'images':bytesList(images_raw)
    }))
    writer.write(example.SerializeToString(