TFRecord 读取图像和标签

#-*- coding:utf-8 -*-
import tensorflow as tf
from PIL import Image

cwd = 'f:/py/tfrecord/aa/'          # 读取图片后存放的文件路径
filename = tf.train.string_input_producer(['f:/py/tfrecord/train.tfrecords'])
reader = tf.TFRecordReader()          # 建立reader
_,serializer = reader.read(filename)  # 读取文件

# 以下是需要读取的的内容，key与存放时的要一致，tf.FixedLenFeature([],tf.string)的tf.string也要与存放时的一致
feature = tf.parse_single_example(serializer,features={'label':tf.FixedLenFeature([],tf.string),
                                                       'img_raw':tf.FixedLenFeature([],tf.string),
                                                       'img_w': tf.FixedLenFeature([], tf.int64),
                                                       'img_h': tf.FixedLenFeature([], tf.int64),
                                                       'img_c': tf.FixedLenFeature([], tf.int64)})

'''
# img取出的格式是string，需要转换为tf.uint8,这五个参数都只是设定好，还没实际运行。
# 如果取出图片是统一的shape，就可以在
'''
img = tf.decode_raw(feature['img_raw'],tf.uint8)
img_w = feature['img_w']    #图像的宽，int
img_h = feature['img_h']    #图像的高，int
img_c = feature['img_c']    #图像的通道数，int
label = feature['label']    #图像的标签，bytes    输出为 b'japandog'，所以下面需要decode
# img = tf.reshape(img,[256,256,3]) 如果想要固定图片的shape，就可以在这里添加这句，如果要原图的shape，只能在取出img_w,img_h,img_c之后，再使用tf.reshape(),否则会报错。
with tf.Session() as sess: #开始一个会话
    coord=tf.train.Coordinator()
    threads= tf.train.start_queue_runners(coord=coord)
    for i in range(10):
        example,w,h,c,label_out = sess.run([img,img_w,img_h,img_c,label])
        label_out=label_out.decode('utf-8')
        img_new = sess.run(tf.reshape(example, [w,h,c])) # tf.reshape需要sess.run()
        img_show=Image.fromarray(img_new)
        img_show.save(cwd+str(i)+label_out+'.jpg')
    coord.request_stop()
    coord.join(threads)