PyTorch 貓狗分類源代碼及數(shù)據(jù)集

如果你對(duì)圖像識(shí)別學(xué)習(xí)感興趣，可以試試這個(gè)小項(xiàng)目-PyTorch 貓狗分類。

首先就是安裝 PyTorch

pip3 install torch torchvision

然后準(zhǔn)備數(shù)據(jù)集，訓(xùn)練，驗(yàn)證。

這里有一份代碼和數(shù)據(jù)集，你下載后，在目錄內(nèi)啟動(dòng) jupyter notebook 就可以直接運(yùn)行。

400 張圖片，10 輪訓(xùn)練后的準(zhǔn)確率達(dá)到 94.25%，你可以進(jìn)一步調(diào)整參數(shù)進(jìn)行優(yōu)化：

源代碼及數(shù)據(jù)集我打包放在了百度網(wǎng)盤，公眾號(hào)「Python七號(hào)」回復(fù)「貓狗識(shí)別」獲取。