无意间接触到利用卷积神经网络来进行手写体识别，觉得好玩便尝试一下

版本：python 3.8 Tensorflow 2.9

安装Tensorflow

首先我们需要安装Tensorflow,我这里是在anaconda中新建了一个虚拟环境，名叫tensorflow(另外说一句anaconda简直是太好用了！)

conda create --name tensorflow python=3.8

安装完之后，默认在tensorflow虚拟环境里面，如果不在，在anaconda的shell中 输入activate tensorflow 进入tensorflow环境

现在，我们来安装tensorflow

pip install tensorflow==2.9 -i https://pypi.doubanio.com/simple/

这里我是安装的2.9版本，大家可以自行选择。

然后安装 tensorflow 所需要的包

pip install pandas matplotlib notebook -i https://pypi.doubanio.com/simple/

这里也是根据自己的选择来安装

MINIST数据集

我们都知道，要训练一个模型，需要数据集的支持，而tensorflow提供了MINIST数据集，分为数据集和测试集。

训练集: 有6万张图片,每张图片大小28x28,同时有对应数量的标签(就是每张图片对应的数字)

测试集: 与训练集相比,其他一样，就是图片只有1万张

我们可以看看测试集的大小与部分图片 (注意 这里必须安装matplotlib库)

import tensorflow as tf
import matplotlib.pyplot as plt
 
# 导入数据集
mnist = tf.keras.datasets.mnist
(train_images, train_labels), (test_images, test_labels) = mnist.load_data()
print("训练集样本及标签", train_images.shape, train_labels.shape)
print("测试集样本及标签", test_images.shape, test_labels.shape)
train_images, test_images = train_images / 255.0, test_images / 255.0  # 归一化,不然梯度爆炸
# 进行绘画
for i in range(15):
    plt.subplot(3, 5, i + 1)
    plt.xticks([])
    plt.yticks([])
    plt.grid(False)
    plt.imshow(test_images[i])
    plt.xlabel(test_labels[i])
plt.show()

差不多是下面这个样子

训练CNN卷积神经网络

以下是对代码所定义的神经网络模型的讲解（虽然不是我写的）

1. 第一层为卷积层,32卷积核(即32输出通道),卷积核大小5x5,激活函数Relu,零值等大填充
2. 第二层为池化层，2x2最大值池化
3. 第三层为卷积层，卷积核大小5x5，64个输出通道
4. 第四层为池化层，仍为2x2最大池化
5. 第五层为扁平化层，将输入的二维张量拉成一维，便于输入给全连接层
6. 第六层为全连接层，与书上不同，这里是64个神经元(因为我的电脑内存不大)
7. 第七层为Dropout层，随机丢掉一些全连接层的神经元，以避免过拟合
8. 第八层，即最后一层，为输出层，输出独热编码

import tensorflow as tf
 
"""
当前版本:python-3.8,tensorflow-2.9
"""
 
# 导入数据集
mnist = tf.keras.datasets.mnist
(train_images, train_labels), (test_images, test_labels) = mnist.load_data()
print("训练集样本及标签", train_images.shape, train_labels.shape)
print("测试集样本及标签", test_images.shape, test_labels.shape)
train_images, test_images = train_images / 255.0, test_images / 255.0  # 归一化,不然梯度爆炸
 
# 建立各层神经网络
model = tf.keras.models.Sequential()  # 建立一个堆叠层的神经网络
# 第一卷积层,32卷积核(即32输出通道),卷积核大小5x5,使用Relu激活函数,零值等大填充,输入张量形状28x28,色彩通道为1(即黑白图片)
model.add(tf.keras.layers.Conv2D(32, (5, 5), activation='relu', padding='same', input_shape=(28, 28, 1)))
# 2x2的最大值池化
model.add(tf.keras.layers.MaxPooling2D((2, 2)))
# 第二卷积层,64个输出通道，输入通道这里就不用指定，可以自动承接前一层的
model.add(tf.keras.layers.Conv2D(64, (5, 5), activation='relu', padding='same'))
model.add(tf.keras.layers.MaxPooling2D((2, 2)))
# 扁平化,将二维的张量变成一维,这里28x28经过两次2x2池化,已是7x7大小,现在变成49
model.add(tf.keras.layers.Flatten())
# 全连接层,64个神经元
model.add(tf.keras.layers.Dense(64, activation='relu'))
# dropout层，损失函数0.5
model.add(tf.keras.layers.Dropout(0.5))
# Readout层，输出独热编码
model.add(tf.keras.layers.Dense(10))  # 最后输出10个数
 
# 编译模型
model.compile(optimizer='adam',  # Adam优化器
              loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),  # 损失函数
              metrics=['accuracy'])  # 监控指标:精度
# 开始训练,训练周期8,即将所有训练样本(6万个),遍历八遍,因为输入通道是32个,所以每遍训练1875次,每次32个
model.fit(train_images, train_labels, epochs=8, validation_data=(test_images, test_labels))
 
# 训练完毕，使用测试集来评估模型精度
test_loss, test_acc = model.evaluate(test_images, test_labels, verbose=2)
print('\n最终测试集上的精度为:', test_acc)
 
# 保存模型
model.save("../CNN模型")  # 这是自定义的路径,删除"/兼收并蓄"即可直接运行

现在来让我们看一下运行结果：

使用训练好的模型进行预测

使用MINIST数据集的测试集

import tensorflow as tf
 
# 导入数据集
mnist = tf.keras.datasets.mnist
(train_images, train_labels), (test_images, test_labels) = mnist.load_data()
# 加入待预测的图像及其标签
show_images, show_labels = test_images[0:20], test_labels[0:20]
print("所选取的测试集图片的标签is:", show_labels)
# 引入模型
model = tf.keras.models.load_model("../CNN模型")
# 开始预测
predictions = model.predict(show_images)
# 预测完成
predictions_num = tf.argmax(predictions, 1)
# 输出预测标签
print("所预测的标签为:", predictions_num)

运行结果

我们可以看到，全都预测对了

识手写数字识别

接下来让我们来试试自己写的数字能不能识别

这里我们用画图来画一个28*28的黑底白字的png图

这里我画了一个数字4，然后让我们带入代码中，看看能不能识别出来。

import tensorflow as tf

# 读取一张自己手写的图片
img_01 = tf.io.read_file("C:/Users/31849/Desktop/数字识别.png")  # 要黑白图片路径
img02 = tf.io.decode_png(img_01, channels=1)
img03 = tf.image.resize(img02, [28, 28])
img = (img03.numpy()).reshape([1, 28, 28])
print("图片转化之后的形状为:", img.shape)
# 运用模型进行预测
model = tf.keras.models.load_model("../CNN模型")
predictions = model.predict(img)
predictions_num = tf.argmax(predictions, 1)
print("你的模型认为该数字为:", predictions_num.numpy())

运行结果：

对了！ 这说明我们训练的模型有用！

参考文献

• 基于CNn的MINIST手写体识别| (CSDN)
• 血泪经验！总结安装tensorflow库全教程 | (CSDN)