1. 程式人生 > >21個專案玩轉深度學習---自我解析

21個專案玩轉深度學習---自我解析

入門深度學習後,《21個專案玩轉深度學習》將對我在實踐中有很大的提升,在提升過程中將持續更新相關專案內容。

我是從第三章《打造自己的影象識別模型開始》

      在這一章,主要關注的是在自己的影象資料上訓練深度學習模型,主要有:fine tune。其具體的過程分為四步:

1)資料準備

2)訓練模型

3)在測試集 上驗證準確率

4)匯出模型並對單張圖片分類

將VGG16用於其他的資料集上,首先要將fc8這一層去掉,新的輸出層的輸出類別數是我們的資料集的類別數。

初始化模型時使用之前訓練好的網路引數來初始化,這樣訓練時間會大大減少,而且只需要訓練有限的層數。

一般只需要訓練fc8.

1)資料準