MNIST数据集是由0-9,10个手写数字组成。训练图像有60000张,测试图像有10000张。
1、在tensorflow中可以使用python下载数据集,调用download.py。代码如下:
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot = True)
MNIST_data中数据不存在时,则自动将MNIST数据下载到该文件夹下。
one_hot表示独热码,一种类似于二进制的编码,例如0-9一共10个数,那么独热码就为10位:
0
1 0 0 0 0 0 0 0 0 0
1
0 1 0 0 0 0 0 0 0 0
2
0 0 1 0 0 0 0 0 0 0
3
0 0 0 1 0 0 0 0 0 0
4
0 0 0 0 1 0 0 0 0 0
5
0 0 0 0 0 1 0 0 0 0
6
0 0 0 0 0 0 1 0 0 0
7
0 0 0 0 0 0 0 1 0 0
8
0 0 0 0 0 0 0 0 1 0
9
0 0 0 0 0 0 0 0 0 1
2、定义卷积计算函数
def weight_variable(shape):
initial = tf.truncated_normal(shape, stddev = 0.1)