原來在一本書上看到過，從多個tfrecord檔案中讀取資料的方法，今天想用在網上找了一下，現在記錄一下，免得自己以後忘記了又不好找，

tfrecord_file_path = '/train/*.tfrecords'#train是存放tfrecord的資料夾
filename_queue = tf.train.string_input_producer(
               tf.train.match_filenames_once(tfrecord_file_path),shuffle=True,num_epochs=None) #None表示沒喲限制
reader = tf.TFRecordReader()
_,serialized_example = reader.read(filename_queue)  #返回檔名和檔案
features = tf.parse_single_example(serialized_example,features={XXXXXXX}) #取出XXXXXXX

以上這篇從多個tfrecord檔案中無限讀取檔案的例子就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。