貓狗大戰挑戰賽
在 ImageNet 上預訓練 的 VGG 網絡進行測試。因為原網絡的分類結果是1000類,是以這裡進行遷移學習,對原網絡進行 fine-tune (即固定前面若幹層,作為特征提取器,隻重新訓練最後兩層)。
訓練代碼
測試代碼
貓狗大戰挑戰賽
在 ImageNet 上預訓練 的 VGG 網絡進行測試。因為原網絡的分類結果是1000類,是以這裡進行遷移學習,對原網絡進行 fine-tune (即固定前面若幹層,作為特征提取器,隻重新訓練最後兩層)。
訓練代碼
測試代碼