今天在学习tf2.0版本小练习的时候发现fashion_mnist数据集无法正常从官方给定的url中下载,也就是load_data方法。所以采用了手动下载手动加载修改load_data源码方式
首先从这里下载训练数据和测试数据到本地
,将下载好的四个压缩包放在本地项目路径下,然后进入load_data的源码(这里也可以把源码的这个方法拉到自己的脚本文件下,我在这里直接对源码进行了修改)。
主要修改如下几个地方:
# dirname = os.path.join('datasets', 'fashion-mnist')
base = 'fashion_mnist/'
files = [
'train-labels-idx1-ubyte.gz', 'train-images-idx3-ubyte.gz',
't10k-labels-idx1-ubyte.gz', 't10k-images-idx3-ubyte.gz'
]
paths = []
for fname in files:
paths.append(base+fname)
将dirname注释掉,然后将base修改为数据集存放的文件路径,也就是自己项目的文件下,paths数组直接对修改的路径添加,修改为上面的路径,之后再使用load_data方法就直接从本地加载数据集了。
有需要的可以对照上面的代码稍微修改一下就行了。