TensorFlow 提供了一个库, 可以直接用来自动下载与安装MNIST ,安装代码如下:
from tensorflow.examples.tutorials.mnist import input_data #通过包引入类
mnist = input_data.read_data_sets
运行该段代码时会出现如下报错:
ModuleNotFoundError: No module named 'tensorflow.examples.tutorials'
网上有博客提供的一个方法,代码第一行是下载MNIST
文件,网上有提供修改DEFAULT_SOURCE_URL
地址,有好几篇博客说是在mnist.py
文件中进行修改,我是在input_data.py
这个文件里面没有找到DEFAULT_SOURCE_URL
的。
原路径:
# CVDF mirror of http://yann.lecun.com/exdb/mnist/
DEFAULT_SOURCE_URL = 'https://storage.googleapis.com/cvdf-datasets/mnist/'
修改为:
# CVDF mirror of http://yann.lecun.com/exdb/mnist/
# DEFAULT_SOURCE_URL = 'https://storage.googleapis.com/cvdf-datasets/mnist/'
DEFAULT_SOURCE_URL = 'http://yann.lecun.com/exdb/mnist/'
但测试后还不行,于是想到缺文件,就把MNIST文件补上,看可不可以。
将MNIST_data
文件夹放在运行目录下 ,测试下面代码,可以完美运行。
# -*- coding: utf-8 -*-
from tensorflow.examples.tutorials.mnist import input_data #通过包引入类
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True) #编码为one_hot格式
print ('输入数据:',mnist.train.images)
print ('输入数据打shape:',mnist.train.images.shape)
import pylab
im = mnist.train.images[1] #取第1张图片
im = im.reshape(-1,28) #转换为28*28尺寸
pylab.imshow(im)
pylab.show()
print ('输入数据打shape:',mnist.test.images.shape) #测试集
print ('输入数据打shape:',mnist.validation.images.shape) #验证集
MNIST文件下载链接:
CSDN : MNIST_data
没有积分的:
链接:https://pan.baidu.com/s/1Fg3hkG547xg3Rr9K2CPcnQ
提取码:kjmy