python3下tensorflow練習(八)之遷移學習

 這周幫同學做了一個CNN的分類任務,因爲趕時間所以直接用InceptionV3的參數進行遷移學習,只替換最後一層全連接層,然後對自己的數據集進行4分類的訓練。在最後這一層全連接層之前的網絡層稱之爲瓶頸層(BOTTLENECK)。 最後實現了對'雨凇','霜','積雪', '露'的四分類,需要先下載InceptionV3的訓練好的模型和我的四分類數據集 1.訓練的代碼 # -*- coding:
相關文章
相關標籤/搜索