這裏的string_tensor直接寫入“文件名、路徑”所構成的列表即可,tensorflow自動將其轉換爲一階張量;shuffle=True作用爲打亂傳入的文件名的順序。
# 1、構造一個文件名隊列
file_queue = tf.train.string_input_producer(string_tensor,shuffle=True)
# 2、讀取與解碼
# 先實例化讀取器與解碼器,舉例如下:
# 文本文件
file_reader = tf.TextLineReader()
# 圖片文件
file_reader = tf.WholeFileReader()
# 二進制文件
file_reader = tf.FixedLengthRecordReader(record_bytes)
# TFRecords文件
file_reader = tf.TFRecordReader()
# 之後用實例化的文件讀取器調用read(file_queue)方法進行文件讀取
key, value = file_reader.read(file_queue)
# 文本文件解碼
tf.decode_csv()
# 圖片文件解碼
tf.image.decode_jpeg(contents)
tf.image.decode_png(contents)
# 二進制文件解碼
tf.decode_raw()