TensorFlow學習記錄-- 7.TensorFlow高效讀取數據之tfrecord詳細解讀

一 why tfrecord? 對於數據量較小而言,可能通常選擇直接將數據加載進內存,而後再分batch輸入網絡進行訓練(tip:使用這種方法時,結合yield 使用更爲簡潔,你們本身嘗試一下吧,我就不贅述了)。可是,若是數據量較大,這樣的方法就不適用了,由於太耗內存,因此這時最好使用tensorflow提供的隊列queue,也就是第二種方法 從文件讀取數據。對於一些特定的讀取,好比csv文件格式
相關文章
相關標籤/搜索