Keras简介：

Keras是由纯python编写的基于theano/tensorflow的深度学习框架。

Keras是一个高层神经网络API，支持快速实验，能够把你的idea迅速转换为结果，如果有如下需求，可以优先选择Keras：

简易和快速的原型设计（keras具有高度模块化，极简，和可扩充特性）
支持CNN和RNN，或二者的结合
无缝CPU和GPU切换

Keras官网首页

一、背景

本次构建神经网络最终目的：输入一张手写数字图片后，网络输出该图片对应的数字。

二、Keras代码实现

import tensorflow as tf
from tensorflow.keras.datasets import mnist

如果没有安装TensorFlow，那么可以参考我之前的文章：重装CUDA和cuDNN(目的是装TensorFlow和pytorch)【个人梳理总结】

我们使用Keras下的MNIST手写字符数据集，可以使用如下命令下载数据集：

(train_images, train_labels), (test_images, test_labels) = keras.datasets.mnist.load_data()
'''
load_data() returns tuple of NumPy arrays: (x_train, y_train), (x_test, y_test).
train_images: uint8 NumPy array of grayscale image data with shapes (60000, 28, 28), containing the training data.
Pixel values range from 0 to 255.
train_labels: uint8 NumPy array of digit labels (integers in range 0-9) with shape (60000,) for the training data.
test_images: uint8 NumPy array of grayscale image data with shapes (10000, 28, 28), containing the test data.
Pixel values range from 0 to 255.
test_labels: uint8 NumPy array of digit labels (integers in range 0-9) with shape (10000,) for the test data.
'''

下面是选做步骤~

如果想验证一下各个变量的shape可以使用python中的断言关键字：

assert train_images.shape == (60000, 28, 28)
assert train_labels.shape == (10000, 28, 28)
assert test_images.shape == (60000,)
assert test_labels.shape == (10000,)

我们可以使用matplotlib先为我们显示一下测试集第一张图片，请记住它：

digit = test_images[0]
import matplotlib.pyplot as plt

plt.imshow(digit, cmap=plt.cm.binary)  # 输出二值化图像
plt.show()

需要安装matplotlib，请移步至Matplotlib 3.5.2 documentation

或者可以打印test_labels，记住第一个元素的label是什么。

print('test_labels', test_labels)

不出意外的话结果是“7”。

Sequential序贯模型是多个网络层的线性堆叠，也就是“一条路走到黑”。

可以通过向Sequential模型传递一个layer的list来构造该模型，也可以像本文使用的通过.add()方法一个个的将layer加入模型中：

from tensorflow.keras import models
from tensorflow.keras import layers

network = models.Sequential()  # 创建实例命名为network。Sequential意为顺序的，即序贯模型。
network.add(layers.Dense(512, activation='relu', input_shape=(28*28,)))  # 第一层需要加一个input_shape关键字参数
network.add(layers.Dense(10, activation='softmax'))  # ①输出层的units=10，意味着存在10个类别，实际意义为输出的结果是从0~10这是个数字。②我们想要将结果的分类数值范围限定在[0,1]之间，理论上activation也可以换成其他能够将结果限定在[0,1]的激活函数

关于Dense层的理解，课阅读深入理解 KERAS 中 DENSE 层参数

此时，我们可以通过一些命令整体查看搭建的神经网络(可任选其一)：

network.summary()  # 输出一下搭建的神经网络框架总结
tf.keras.utils.plot_model(network, "my_first_network.png")  # 图形化输出
keras.utils.plot_model(network, "my_first_network_with_shape_info.png", show_shapes=True)  # 带有输入shape和输出shape的图形化输出

输出结果省略，各位可在实践中自行感受。

完成模型的搭建后，我们需要使用.compile()方法来编译模型：

优化器optimizer：该参数可指定为已预定义的优化器名，如rmsprop、adagrad，或一个Optimizer类的对象；
损失函数loss：该参数为模型试图最小化的目标函数，它可为预定义的损失函数名，如categorical_crossentropy、mse，也可以为一个损失函数；
指标列表metrics：对分类问题，我们一般将该列表设置为metrics=['accuracy']。指标可以是一个预定义指标的名字,也可以是一个用户定制的函数。指标函数应该返回单个张量,或一个完成metric_name - > metric_value映射的字典。

network.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['accuracy'])

2.4.1 对训练集和测试集的数据部分预处理

将每张图片数据由(28， 28)的二维数组变成(28 * 28)的一维数组
将灰度图中[0, 255]的整数灰度值归一化为[0, 1]的浮点数

train_images = train_images.reshape((60000, 28 * 28))
train_images = train_images.astype('float32') / 255
test_images = test_images.reshape((10000, 28 * 28))
test_images = test_images.astype('float32') / 255

2.4.2 对训练集和测试集的标签部分预处理

对应输出层的units=10，将标签做one-hot编码，用一个拥有10个元素的一位数组替换标签。我们需要把数值7变成一个含有10个元素的数组，然后在第8个元素设置为1，其他元素设置为0，即[0., 0., 0., 0., 0., 0., 0., 1., 0., 0.]

from tensorflow.keras.utils import to_categorical

print("before change:", test_labels[0])
train_labels = to_categorical(train_labels)
test_labels = to_categorical(test_labels)
print("after change: ", test_labels[0])

你一定发现了我们使用了to_categorical函数。它是将类别向量(从0到nb_classes的整数向量)映射为二值类别矩阵, 用于应用到以categorical_crossentropy为目标函数的模型中.

而我们在使用.compile()方法来编译模型参数中损失函数loss='categorical_crossentropy'。

network.fit(train_images, train_labels, epochs=5, batch_size=128)
"""
传入fit()各变量的含义：
train_images：用于训练的手写数字图片；
train_labels：对应的是图片的标记；
batch_size=128：每次网络从输入的图片数组中随机选取128个作为一组进行计算。
epochs=5: 每次计算训练数据将会被遍历5次。
fit函数返回一个History的对象，其History.history属性记录了损失函数和其他指标的数值随epoch变化的情况，如果有验证集的话，也包含了验证集的这些指标变化情况。
"""

测试数据输入，检验网络学习后的图片识别效果。

P.S. 识别效果与硬件有关(CPU/GPU)

本人使用的TensorFlow 2.5搭配GPU为：NVIDIA GeForce GTX 1660 Ti with Max-Q Design computeCapability: 7.5(辣鸡啊！多么想要一个3080！！！)

test_loss, test_acc = network.evaluate(test_images, test_labels, verbose=1)
# verbose: 日志记录——0:静默不显示任何信息，1(default):输出进度条记录
print('test_loss', test_loss)  # 打印loss
print('test_acc', test_acc)  # 打印accuracy

展示训练过程输出信息：

随机输入一张手写数字图片到网络中，看看它的识别效果

(train_images, train_labels), (test_images, test_labels) = mnist.load_data()
digit = test_images[1]  # 挑选测试集第2张图片，是数字2

plt.imshow(digit, cmap=plt.cm.binary)
plt.show()

test_images = test_images.reshape((10000, 28*28))
res = network.predict(test_images)  # 应用已经训练好的模型进行预测

for i in range(res[1].shape[0]):  # 提取第二个预测结果，即对数字2的预测结果
    if res[1][i] == 1:  # 看结果中10个元素的数组中第几位是1
        print("the number for the picture is : ", i)  # 第几位是1就输出这个数字的预测结果是几
        break

展示预测结果输出：