pytorch——用resnet18作動物多分類問題（含可視化結果）

代碼以及數據集下載：https://github.com/duchp/python-all/tree/master/CV%20code/動物多分類項目python 1、任務介紹 綱分類問題，預測該動物是屬於哺乳綱(Mammals)仍是鳥綱(Birds) 種分類問題，預測該動物是兔子、老鼠仍是雞 多任務分類問題，同時預測動物的「綱」和「種」 2、數據預處理 使用的數據集結構以下：git 上圖爲文件夾

>>阅读原文<<