在用Pytorch实现手写数字识别中加载本地mnist库的方法

在使用mnist库的过程中,如果选择download=True,就会在F5之后自动从网上下载mnist库,这个过程十分缓慢切容易导致timeerror,所以就先从网上下载mnist文件再本地加载。

下载网址:
http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz

http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz

http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz

http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz

下载后,是如下图所示的四个文件:
在这里插入图片描述
解压缩不用多说
然后自己建一个mnist文件夹,为了方便我就和 .py建在一起
在这里插入图片描述
然后在mnist中再建一个MNIST文件夹
在这里插入图片描述
在其中在建两个文件夹分别叫processed和raw
processed中是用来存放训练数据的不用管
在这里插入图片描述
在raw中放入刚刚解压缩的四个文件
在用Pytorch实现手写数字识别中加载本地mnist库的方法_第1张图片
对应的在.py中的code形式是
在这里插入图片描述
其中root的内容根据你的mnist文件存放的位置而定
综上所述就能在本地加载mnist库了

你可能感兴趣的:(小Tip)