深度學習tensorflow實戰筆記(3)VGG-16訓練本身的數據並測試和保存模型

    前面的博客介紹瞭如何把圖像數據轉換成tfrecords格式並讀取,本篇博客介紹如何用本身的tfrecords格式的數據訓練CNN模型,採用的模型是VGG-16。現有的教程都是在mnist或者cifar-10數據集上作的訓練,如何用本身的數據集進行訓練相關的資料比較少,即便有相關的也是不完整,因此作一個完成流程的學習筆記,以供之後查閱使用。首先要作的是把本身的數據轉換成tfrecords文件
相關文章
相關標籤/搜索