天天看點

貓狗大戰

貓狗大戰挑戰賽

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

訓練代碼

測試代碼