2017-08-07 30 views
0

我有3個班。 (訓練50k,驗證12k) 通過使用預訓練的vgg16和resnet50,凍結模型並僅在頂部訓練密集層,我的驗證準確率達到99%。 我應該微調以通過解凍圖層來改進功能還是應該使用這些功能? 另外,vgg16是比Resnet50更好的特徵提取程序,還是應該使用Resnet中的功能? 謝謝!如何改進convnnet的圖像檢索功能?

回答

0

這取決於您的問題域。如果您正在對相同問題域的預訓練模型進行微調,並且訓練數據量很小,那麼您所做的是正確的。

也許如果你只凍結了第一層,這些層對於一般特徵提取(egdes,blob,shapes ..等)有很好的訓練,你可以提高你的性能。它還建議申請數據增強,如果你要做到這一點,以避免過度擬合

我鼓勵你去檢查遷移學習下面的教程瞭解更多詳情: http://cs231n.github.io/transfer-learning/