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)