用卷积神经网络对mnist进行数字识别程序（tensorflow）

最新推荐文章于 2024-08-02 12:58:05 发布

忧郁一休

最新推荐文章于 2024-08-02 12:58:05 发布

阅读量1.6k

点赞数 2

分类专栏： tensorflow 卷积神经网络神经网络文章标签：卷积神经网络 CNN MNIST tensorflow 手写数字识别

本文链接：https://blog.csdn.net/youyuyixiu/article/details/71269725

版权

#下载数据集
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data",one_hot = True)

#引入tensorflow
import tensorflow as tf
#建立session对象
sess = tf.InteractiveSession()

#占位符（图像和标签）
x = tf.placeholder(tf.float32, shape=[None, 784])
y_ = tf.placeholder(tf.float32, shape=[None, 10])

#权重函数
def weight_variable(shape):
    initial = tf.truncated_normal(shape,stddev = 0.1)
    return tf.Variable(initial)
#偏置函数
def bias_variable(shape):
    initial = tf.constant(0.1,shape = shape)
    return tf.Variable(initial)
#卷积函数
def conv2d(x,W):
    return tf.nn.conv2d(x, W, strides = [1,1,1,1], padding = "SAME")
#池化函数
def max_pool_2x2(x):
    return tf.nn.max_pool(x, ksize = [