tensorflow如何訓練自己的圖像數據?
最近看了一些tensorflow的教程,跟著做了一些練習。例子中都是已經做好的數據了,想快點用自己的圖像數據做個訓練,看看結果。哪位前輩能夠給一點建議,或者是類似的教程嗎?謝謝謝謝
你如果玩tensorflow的話可以使用tfrecord,How do I convert a directory of jpeg images to TFRecords file in tensorflow?這裡面有個例子怎麼做tfrecord的生成,另外如果你不想這樣準備的話可以用tflearn,很好用的一個框架:
tflearn.data_utils.image_preloader (target_path, image_shape, mode="file", normalize=True, grayscale=False, categorical_labels=True, files_extension=None, filter_channel=False)
可以有兩種mode 一種file,直接有一個文件保存label和path信息,一種是folder,直接讀目錄,子目錄是label,目錄文件為對應label下的圖像文件
可以使用 tensorflow 提供的 batch generator, 首先把數據送入隊列中 queue, 然後需要一種 tensorflow reader (tf 有好幾種reader), 然後從隊列中讀取數據, 然後使用 tf.train.batch 或者 tf.train.shuffle_batch 就可以生成你定義的 batch size 的數據了,需要用 Coordinator()和 start_queue_runner 監控隊列的狀態。
我製作了一個 tensorflow 實戰教程,第一系列就是:
如果使用tensorflow處理我們自己的數據?
比如一堆圖片存放在本地硬碟上,
怎麼讀取數據,生成batch,train?
怎麼保存訓練好的模型結構和參數?
怎麼恢復訓練好的結構和參數,並使用新的數據測試?
怎麼測試單張圖片?
我使用的是Kaggle貓狗大戰的數據,25000張圖片。 鏈接:
優酷: http://i.youku.com/deeplearning101
深度學習QQ群:153032765
youtube:
https://www.youtube.com/channel/UCVCSn4qQXTDAtGWpWAe4Plw
good luck!
可以參考我寫的這篇博客,在訓練時候直接使用圖片數據做為輸入而不是生成好的二進位文件,鏈接:https://mrlittlepig.github.io/2017/04/30/tensorflow-for-image-processing/
建議參考tensorflow中MNIST數據集讀取方式 read_data_sets, 然後自己實現讀取kaggle社區digit recogniizer數據集Digit Recognizer | Kaggle ,我自己實現了一個input_data.py, 放在geekerlw/tensorflow-learn ,不過有瑕疵,僅供參考,手動捂臉
我寫的圖片資料庫/streamer. http://github.com/aaalgo/picpac
推薦閱讀:
※關於Tensorflow的一些想法?
※tensorflow中的tensorboard可視化中的準確率損失率曲線,為什麼有類似毛刺一樣?
※Tensorflow 中怎麼定義自己的層呢?
※怎樣使用tensorflow導入已經下載好的mnist數據集?
※Google I/O 2017 上有哪些關於 TensorFlow 的前沿應用和分享?
TAG:TensorFlow |