1. 程式人生 > >TensorFlow隊列讀取流程

TensorFlow隊列讀取流程

pro 文件 alt flow 單元 狀態 sta shuff img

技術分享

技術分享

主要相關函數:

tf.train.string_input_producer:這個函數需要傳入一個文件名list,系統會自動將它轉為一個文件名隊列

tf.train.string_input_producer:兩個重要的參數(1)num_epochs,上文中提到的epoch數(2)shuffle,shuffle是指在一個epoch內文件的順序是否被打亂

tf.train.string_input_producer:創建文件名隊列後,整個系統其實還是處於“停滯狀態”的

tf.train.start_queue_runners:啟動填充隊列的線程,這時系統就不再“停滯”。此後計算單元就可以拿到數據並進行計算

TensorFlow隊列讀取流程