1. 程式人生 > >從多個tfrecord檔案中讀取資料的方法

從多個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