卷积神经网络的应用实例-LeNet5网络结构

#encoding:utf-8
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
import time
#获取mnist数据
mnist_data_set = input_data.read_data_sets('MNIST_data', one_hot=True)


#声明输入图片数据，类别
x = tf.placeholder('float',[None,784])
y_ = tf.placeholder('float',[None, 10])
x_image = tf.reshape(x, [-1, 28, 28, 1])

#第一层卷积层，初始化卷积核参数，偏置值，该卷积层5*5大小，一个通道，共有6个不同卷积核
filter1 = tf.Variable(tf.truncated_normal([5,5,1,6]))
bias1 = tf.Variable(tf.truncated_normal([6]))
conv1 = tf.nn.conv2d(x_image, filter1, strides=[1,1,1,1], padding="SAME")
h_conv1 = tf.nn.sigmoid(conv1 + bias1)

maxpool2 = tf.nn.max_pool(h_conv1, ksize=[1,2,2,1], strides=[1,2,2,1],padding='SAME')

filter2 = tf.Variable(tf.truncated_normal([5,5,6,16]))
bias2 = tf.Variable(tf.truncated_normal([16]))
conv2 = tf.nn.conv2d(maxpool2, filter2, s