手写数字识别数据集简介  MNIST数据集（修改的国家标准与技术研究所——Modified National Institute of Standards and Technology），是一个大型的包含手写数字图片的数据集。该数据集由0-9手写数字图片组成，共10个类别。每张图片的大小为28 * 28。  MNIST数据集共有70000张图像，其中训练集60000张，测试集10000张。训练集分为55000张训练图像与5000张验证图像。MNIST图像为单通道。 使用CNN来实现  卷积神经网络：CNN（Convolutional neural network），即卷积神经网络。卷积理解为一个信号与另外一个信号进行叠加，产生新的信号的过程。（在卷积神经网络中，可认为具有固定权重的滑动窗口与原窗口的数据进行对位相乘再相加的过程。）CNN主要层次：   1.输入层------输入的是图片。   2.卷积层------有几组卷积和就有几个输出通道,卷积之后能改变输入的大小。   3.激励层------卷积之后进行激活，激励不会改变大小。   4.池化层------分为最大池化(取最大值)和平均池化（取平均值）。对激励之后的结果进行融合，也会有一个滑动窗口，与经过激励层的结果进行叠加，叠加后取最大值或平均值；经过池化后矩阵变小，相当于变相的减少神经元的个数。以此降低模型的复杂度，防止出现过拟合。池化不会改变通道数。   功能：提取有效特征，降低模型复杂度，防止过拟合。   5.全连接层-----对提取的局部特征信息进行融合构成一个完整的图像。（全连接层不设置连接层的个数，可以为一层，也可以为两层。）   卷积神经网络为了防止过拟合在全连接层设置了一个神经元的随机失活率。 |
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
if __name__ == '__main__':
# 读入数据。
mnist = input_data.read_data_sets("data/", one_hot=True)
with tf.name_scope("input"):
# 训练图像的占位符。
x = tf.placeholder(tf.float32, [None, 784])
# 训练图像对应分类（标签）的占位符。
y = tf.placeholder(tf.float32, [None, 10])
# 因为卷积要求输入的是4维数据，因此对形状进行转换。
# NHWC(默认) NCHW
# N number样本的数量
# H height图像的高度
# W width图像的宽度
# C channel图像的通道数
x_image = tf.reshape(x, [-1, 28, 28, 1])
# 卷积层1。
with tf.name_scope("conv_layer1"):
# 定义权重。（w就是滑动窗口）
# 5, 5, 1, 32 => 滑动窗口的高度，滑动窗口的宽度，输入通道数，输出通道数。
w = tf.Variable(tf.truncated_normal([5, 5, 1, 32], stddev=0.1), name="w")
# 定义偏置。
b = tf.Variable(tf.constant(0.0, shape=[32]), name="b")
# 进行卷积计算。
# strides=[1, 1, 1, 1] 步幅。针对输入的NHWC定义的增量。
# padding： SAME 与VALID。SAME，只要滑动窗口不全移除输入区域就可以。
# VALID，滑动窗口必须完全在输入区域之内。
conv = tf.nn.bias_add(tf.nn.conv2d(x_image, w, strides=[1, 1, 1, 1], padding='SAME'), b, name="conv")
# 使用激活函数进行激活。
activation = tf.nn.relu(conv)
# 池化操作。
# ksize：池化的窗口。
pool = tf.nn.max_pool(activation, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')
# 卷积层2。
with tf.name_scope("conv_layer2"):
w = tf.Variable(tf.truncated_normal([5, 5, 32, 64], stddev=0.1), name="w")
b = tf.Variable(tf.constant(0.0, shape=[64]), name="b")
conv = tf.nn.bias_add(tf.nn.conv2d(pool, w, strides=[1, 1, 1, 1], padding='SAME'), b, name="conv")
activation = tf.nn.relu(conv)
pool = tf.nn.max_pool(activation, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME')
# 全连接层1。
with tf.name_scope("full_layer1"):
# 7 * 7 * 64
# 原始图像是28 * 28，经过卷积与激励后，没有改变，经过2 * 2池化后，变成 14 * 14。
# 第一层卷积之后结果为14 * 14，经过第二层卷积与激励后，没有改变，经过2 * 2池化后，变成 7 * 7。
# 第二层卷积之后，我们图像的形状为 NHWC => [N, 7, 7, 64]
# 4维变成2二维，将后面三维拉伸成为1维。 =》 [N, 7 * 7 * 64]
w = tf.Variable(tf.truncated_normal([7 * 7 * 64, 1024], stddev=0.1), name="w")
b = tf.Variable(tf.constant(0.0, shape=[1024]), name="b")
# 将第二层卷积之后的结果转换成二维结构。
pool = tf.reshape(pool, [-1, 7 * 7 * 64])
activation = tf.nn.relu(tf.matmul(pool, w) + b)
# 执行dropout（随机丢弃）
keep_prob = tf.placeholder(tf.float32)
# 进行随机丢弃，keep_prob指定神经元的保留率。
drop = tf.nn.dropout(activation, keep_prob)
# 全连接层2。
with tf.name_scope("full_layer2"):
w = tf.Variable(tf.truncated_normal([1024, 10], stddev=0.1), name="w")
b = tf.Variable(tf.constant(0.0, shape=[10]), name="b")
logits = tf.matmul(drop, w) + b
# 损失值与准确率计算层。
with tf.name_scope("compute"):
# 计算损失值。
loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels=y, logits=logits))
# 将损失值加入到tensorboard中。
# tf.summary.scalar('loss',loss)
train_step = tf.train.AdamOptimizer(1e-4).minimize(loss)
# 计算准确率
correct = tf.equal(tf.argmax(logits, 1), tf.argmax(y, 1))
accuracy = tf.reduce_mean(tf.cast(correct, tf.float32))
# tf.summary.scalar('accuracy',accuracy)
#合并所有的summary
# merged = tf.summary.merge_all()
# 创建Session。
with tf.Session() as sess:
# 对全局变量进行初始化。
sess.run(tf.global_variables_initializer())
# train_writer = tf.summary.FileWriter('logs/train',sess.graph)
# test_writer = tf.summary.FileWriter('logs/test',sess.graph)
# 可以尝试更大的次数，可以将准确率提升到99%以上。
for i in range(1, 3001):
batch = mnist.train.next_batch(64)
# 每100步报告一次在验证集上的准确度
if i % 100 == 0:
train_accuracy = accuracy.eval(
feed_dict={x: batch[0], y: batch[1], keep_prob: 1.0})
test_accuracy = accuracy.eval(
feed_dict={x: mnist.test.images[:5000], y: mnist.test.labels[:5000], keep_prob: 1.0})
print(f"step {i}, training accuracy {train_accuracy * 100:.2f}%")
print(f"step {i}, test accuracy {test_accuracy * 100:.2f}%")
train_step.run(feed_dict={x: batch[0], y: batch[1], keep_prob: 0.5})
# 计算并写入训练集计算的结果。
# summary = sess.run(merged,feed_dict={x:batch[0], y:batch[1] ,keep_prob:1.0})
# train_writer.add_summary(summary, i)
# 计算并写入测试集计算的结果。
# summary = sess.run(merged,feed_dict={x:mnist.test.images, y:mnist.test.labels, keep_prob:1.0})
# test_writer.add_summary(summary,i) 部分运行结果：
step 100, training accuracy 81.25%
step 100, test accuracy 85.28%
step 200, training accuracy 90.62%
step 200, test accuracy 91.00%
step 300, training accuracy 100.00%
step 300, test accuracy 92.14%
step 400, training accuracy 96.88%
step 400, test accuracy 93.88%
step 500, training accuracy 93.75%
step 500, test accuracy 94.34%
step 600, training accuracy 98.44%
step 600, test accuracy 94.64%
扫一扫
260
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
