Caffe測試集分類結果相同,全部爲最後一類

現象說明 目標:二分類,訓練過程中驗證集準確率比較高。訓練完成之後對測試集進行測試,測試結果中的類別全部爲1(最後一類)。 Caffe測試集分類結果相同,全部爲最後一類 原因及解決方案 原因:沒有對網絡的第一個卷積層進行初始化。下圖是原來網絡第一個卷積層的參數。 解決方案:將第一個卷積層權重初始化爲「xavier」,偏置初始化爲常量0。
相關文章
相關標籤/搜索