簡單理解TensorFlow textlinereader的例子

import tensorflow as tf
import numpy as np

def main():
    filename_queue = tf.train.string_input_producer(["test.txt", "test2.txt"], num_epochs=1)
    reader = tf.TextLineReader()
    key, value = reader.read(filename_queue)

    with tf.Session() as sess:
        sess.run(tf.initialize_local_variables())
        tf.train.start_queue_runners()
        num_examples = 0
        try:
            while True:
                s_key, s_value = sess.run([key, value])
                print( s_key, s_value)
            num_examples += 1
        except tf.errors.OutOfRangeError:
            print ("There are", num_examples, "examples")

if __name__ == "__main__":
    main()

test.txt test2.txt內容隨意。
其結果會輸出test2.txt test.txt的內容(test2.txt的內容先輸出)

核心是先建立TextlineReader() 然後reader.read(其參數爲queue或tensor) 用string_input_producer來生成queue即可

發佈了34 篇原創文章 · 獲贊 18 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章