初學tensorflow,網上下載了一堆圖片,5分類,每個分類的訓練集有3000張圖片,測驗集100張。總共就是15000訓練集,500測驗集。
圖片預處理都是直接resize成224*224。
cnn網路有5個卷積層,3個全連接層,基本參照vgg的結構簡化了一下。
調整了幾次學習率等等超引數,曲線都基本如此。
這是最后一次的曲線,學習率0.01,batch為20。
猜測是資料集有什么問題,但是檢查了label沒有錯,不知道還有什么沒考慮到。
哎,搞了三天了,顯卡一直在滿載,心痛


uj5u.com熱心網友回復:
loss曲線還是不錯的,不過你的曲線嚴重過擬合了。uj5u.com熱心網友回復:
可能是訓練集的特征不能涵蓋測驗集的特征uj5u.com熱心網友回復:
可進行資料增強轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/54204.html
上一篇:最少貨車數量
