如果你对图像识别学习感兴趣,可以试试这个小项目-PyTorch 猫狗分类。
首先就是安装 PyTorch
然后准备数据集,训练,验证。
这里有一份代码和数据集,你下载后,在目录内启动 jupyter notebook 就可以直接运行。
400 张图片,10 轮训练后的准确率达到 94.25%,你可以进一步调整参数进行优化:
如果你对图像识别学习感兴趣,可以试试这个小项目-PyTorch 猫狗分类。
首先就是安装 PyTorch
然后准备数据集,训练,验证。
这里有一份代码和数据集,你下载后,在目录内启动 jupyter notebook 就可以直接运行。
400 张图片,10 轮训练后的准确率达到 94.25%,你可以进一步调整参数进行优化:
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M