PyTorch 猫狗分类源代码及数据集
Python七号
2022-08-01 09:43
如果你对图像识别学习感兴趣,可以试试这个小项目-PyTorch 猫狗分类。
首先就是安装 PyTorch
pip3 install torch torchvision
然后准备数据集,训练,验证。
这里有一份代码和数据集,你下载后,在目录内启动 jupyter notebook 就可以直接运行。
![](https://filescdn.proginn.com/201dd836257b802e5454b3b4f4a1d618/1769885174f03a8763bc4c59ce1a8ef0.webp)
400 张图片,10 轮训练后的准确率达到 94.25%,你可以进一步调整参数进行优化:
![](https://filescdn.proginn.com/864aa12fbbd192bae6ccbc89da7ecc96/15fb5879481746767bfce4229cb17304.webp)
![](https://filescdn.proginn.com/f72359ebf7d0d94c9adc250fff666aa5/c76c31251fabf0726e2812d8022c03e4.webp)
源代码及数据集我打包放在了百度网盘,公众号「Python七号」回复「猫狗识别」获取。
评论