- 博客(8)
- 收藏
- 关注
原创 P7:马铃薯病害识别(VGG-16复现)
我的GPU显存太小,把batchsize调成1之后仍然无法跑这个模型,解决方案:1.网上租服务器2.调整模型结构。
2025-04-18 19:39:16
592
原创 P6:VGG-16算法-Pytorch实现人脸识别
这里注意,如果在线下载失败,可以复制地址到迅雷,下载到本地,放在提示的文件夹里就行。这里我的gpu内存不够,所以调整batchsize为4,可以正常进行。预测结果是:Angelina Jolie。2.指定图片进行预测。
2025-04-11 20:20:50
904
原创 P5:Pytorch实现运动鞋识别
尝试将epoch改为100,测试正确率上涨到84.2%,但是不会随着epoch增加继续增加了。需要优化模型才能继续增长正确率。2. 指定图片进行预测。预测结果是:adidas。
2025-04-04 18:55:35
688
原创 P3:Pytorch实现天气识别
后续还测试了一下其他照片,发现cloud和shine判断比较模糊,还有很依赖于图片的色调,比较暗的照片很容易判断成rain。我感觉是数据集的问题(数据都太典型了)比较大。模型仍有一点优化空间。使用的测试照片是自己拍的一张图片,我感觉判断的还是很准确的,shine和cloudy较高。这里注意data_dir需要更换成自己的相对地址。调用训练好的模型去识别一张本地的天气图片。在正式训练的最后加入此段代码。设置训练集:数据集=8:2。预测结果: shine。
2025-03-14 20:13:49
1114
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅