手写数字识别模型

任务输入：一系列手写数字图片，其中每张图片都是28x28的像素矩阵。
任务输出：经过了大小归一化和居中处理，输出对应的0~9的数字标签。

---

前提条件

---

import paddle
from paddle.nn import Linear
import paddle.nn.functional as F
import os
import matplotlib.pyplot as plt
import numpy as np

数据处理

train_dataset=paddle.vision.datasets.MNIST(mode='train')

train_data0=np.array(train_dataset[0][0])
train_label_0=np.array(train_dataset[0][1])

import matplotlib.pyplot as plt
plt.figure("Image") 
plt.figure(figsize=(2,2))
plt.imshow(train_data0, cmap=plt.cm.binary)
plt.axis('on') 
plt.title('image') 
plt.show()

print("图像数据形状和对应数据为:", train_data0.shape)
print("图像标签形状和对应数据为:", train_label_0.shape, train_label_0)
print("\n打印第一个batch的第一个图像，对应标签数字为{}".format(train_label_0))

模型设计

class MNIST(paddle.nn.Layer):
    def __init__(self):
        super(MNIST,self).__init__()
        self.fc=paddle.nn.Linear(in_features=784,out_features=1)
        
    def forward(self,inputs):
        outputs=self.fc(inputs)
        return outputs

训练配置

model=MNIST()
def train(model):
    model.train()
    train_loader=paddle.io.DataLoader(paddle.vision.datasets.MNIST(mode='train'),
                                     batch_size=16,
                                     shuffle=True)
    opt=paddle.optimizer.SGD(learning_rate=0.001,parameters=model.parameters())

训练过程

训练过程采用二层循环嵌套方式，训练完成后需要保存模型参数，以便后续使用。

内层循环：负责整个数据集的一次遍历，遍历数据集采用分批次（batch）方式。
外层循环：定义遍历数据集的次数，本次训练中外层循环10次，通过参数EPOCH_NUM设置。

def norm_img(img):
    assert len(img.shape)==3
    batch_size,img_h,img_w=img.shape[0],img.shape[1],img.shape[2]
    img=paddle.reshape(img,[batch_size,img_h*img_w])
    return img
    

import paddle
paddle.vision.set_image_backend('cv2')
model=MNIST()

def train(model):
    model.train()
    train_loader=paddle.io.DataLoader(paddle.vision.datasets.MNIST(mode='train'),
                                     batch_size=16,
                                     shuffle=True)
    
    opt=paddle.optimizer.SGD(learning_rate=0.001,parameters=model.parameters())
    EPOCH_NUM=10
    for epoch in range(EPOCH_NUM):
        for batch_id, data in enumerate(train_loader()):
            images = norm_img(data[0]).astype('float32')
            labels = data[1].astype('float32')
            
           
            predicts = model(images)
            
           
            loss = F.square_error_cost(predicts, labels)
            avg_loss = paddle.mean(loss)
            
            
            if batch_id % 1000 == 0:
                print("epoch_id: {}, batch_id: {}, loss is: {}".format(epoch, batch_id, avg_loss.numpy()))
            
          
            avg_loss.backward()
            opt.step()
            opt.clear_grad()
            
train(model)
paddle.save(model.state_dict(), './mnist.pdparams')

模型测试

模型测试的主要目的是验证训练好的模型是否能正确识别出数字，包括如下四步：

声明实例
加载模型：加载训练过程中保存的模型参数。
灌入数据：将测试样本传入模型，模型的状态设置为校验状态（eval），显式告诉框架我们接下来只会使用前向计算的流程，不会计算梯度和梯度反向传播。
获取预测结果，取整后作为预测标签输出。
在模型测试之前，需要先从’./work/example_0.jpg’文件中读取样例图片，并进行归一化处理。

# 导入图像读取第三方库
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image
# 读取图像
img_path = './work/example_0.jpg'
# 显示图像
im = Image.open('./work/example_0.jpg')
plt.imshow(im)
plt.show()

im = im.convert('L')
print('原始图像shape: ', np.array(im).shape)

im = im.resize((28, 28), Image.ANTIALIAS)
plt.imshow(im)
plt.show()
print("采样后图片shape: ", np.array(im).shape)

# 读取一张本地的样例图片，转变成模型输入的格式
def load_image(img_path):
  
    im = Image.open(img_path).convert('L')
    
    im = im.resize((28, 28), Image.ANTIALIAS)
    im = np.array(im).reshape(1, -1).astype(np.float32)
    # 图像归一化，保持和数据集的数据范围一致
    im = 1 - im / 255
    return im


model = MNIST()
params_file_path = 'mnist.pdparams'
img_path = './work/example_0.jpg'

param_dict = paddle.load(params_file_path)
model.load_dict(param_dict)

model.eval()
tensor_img = load_image(img_path)
result = model(paddle.to_tensor(tensor_img))
print('result',result)

print("本次预测的数字是", result.numpy().astype('int32'))

result Tensor(shape=[1, 1], dtype=float32, place=CPUPlace, stop_gradient=False,
       [[nan]])
本次预测的数字是 [[1]]

从打印结果来看，模型预测出的数字是与实际输出的图片的数字不一致。这里只是验证了一个样本的情况，如果我们尝试更多的样本，可发现许多数字图片识别结果是错误的。因此完全复用房价预测的实验并不适用于手写数字识别任务。

总结

paddlepaddle用起来还是很方便的，不论是定义网络结构还是训练速度，都值得一提，然而我个人的体验中，认为最值得说的是这几点：

1.导入数据方便。这次训练的手写数字识别数据量比较小，但是如果想要添加数据，也非常方便，直接添加到相应目录下。
2.速度快。上面的例子已经证明了paddlepaddle的速度，并且在提升速度的同时，模型准确度也与最优结果相差不多，这对于我们训练海量数据的模型是一个极大的优势啊！