PyTorch 猫狗分类源代码及数据集

Python七号

共 274字,需浏览 1分钟

 · 2022-08-01

如果你对图像识别学习感兴趣,可以试试这个小项目-PyTorch 猫狗分类。

首先就是安装 PyTorch

pip3 install torch torchvision

然后准备数据集,训练,验证。

这里有一份代码和数据集,你下载后,在目录内启动 jupyter notebook 就可以直接运行。

400 张图片,10 轮训练后的准确率达到 94.25%,你可以进一步调整参数进行优化:

源代码及数据集我打包放在了百度网盘,公众号「Python七号」回复「猫狗识别」获取。


浏览 37
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报