今天在看莫凡的torch教程,cnn案例。封装的函数不一样,所以第一次跑demo的时候需要下载数据集。
但是我发现,可能是由于网站的维护问题,是的数据集下载非常非常缓慢,难以忍受。所以在结合源码以及以前下载的matlab环境下使用的数据集。做了以下处理:
数据集:
另外先在目录下新建mnist文件夹,在mnist文件夹下再建raw和processed文件夹。
自己的电脑只是路径的不一样。此时文件夹中是空的,先将数据集(压缩格式)复制到raw中:
然后再console中执行下列代码:
root = r'C:\Users\admin\PycharmProjects\py_projects\pytorch_learning\mnist'
raw_folder='raw'
filenames=['train-images-idx3-ubyte.gz','train-labels-idx1-ubyte.gz','t10k-images-idx3-ubyte.gz','t10k-labels-idx1-ubyte.gz']
import os
import g