![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
小小子596
这个作者很懒,什么都没留下…
展开
-
pytorch 使用AlexNet实现Mnist手写数字识别
AlexNet网络框架如下:AlexNet的原始输入图片大小为224*224,Mnist数据集中图片大小为28*28,所以需要对网络参数进行修改。先掉用train函数进行训练,训练好的参数会保存在params.pth文件中,训练好使用本地图片(画图软件生成)进行测试。完整程序如下:import torchimport torchvision as tvimport torchvi...原创 2020-03-16 16:23:02 · 6332 阅读 · 17 评论 -
pytorch实现个人手写数字识别
网上的大多数例子都是基于Mnist数据集进行测试的,今天实现一个自己手写数字的识别。首先训练模型,使用Mnist数据集,网络的backbone采用LeNet。1. 导入需要的模块并添加GPU设备import torchimport torchvision as tvimport torchvision.transforms as transformsimport torc...原创 2020-03-15 21:14:27 · 967 阅读 · 0 评论