32. 深度学习进阶 - Transfer Learning

Hi，你好。我是茶桁。

之前的课程中，咱们学习了CNN的原理，学习了pooling, fully connected是做什么的。还了解了理论上简单的模型也是可以做事情的，只不过在特定的一些情况下要解决问题的时候简单方法效果不太好，所以用了像LSTM，或者RNN、CNN之类的结构。

这些本质上都是在做特征的提取。一个经典的观念是，神经网络其实一共都可以分成两个部分，第一个部分是特征提取，第二个部分是分类器。像fully connected layer，其实就是之后再加上一个Softmax或者log Softmax，在做分类器的实现。

前面在进入全连接层之前，也在进入Softmax或者log Softmax之前，全部做的都是特征提取的事。

不管你是线性函数，就线性变化全连接的这种网络，还是RNN，LSTM或者CNN等等，在进入Softmax之前，这些都是在做特征提取。

Hinton当时的说法我觉得说的很有道理，就说「特征提取的作用是让相似的东西不相似，让不相似的相似」。

意思就是，我们对于任何一个神经网络来说，到最终的这个全连接，加上Softmax，之前的这些东西不管你是输入的是一个图片还是几个文字，还是说一串数据。所谓的让看起来相似的东西不相似是如果有两个图片，或者两组数据，他们在我们人看起来是比较类似的。但是假设他们的label不一样的话，我们整个特征提取的过程是把输入的这两个x，人看起来是一样的，在最后输出的这个地方要尽可能的不一样。所以送到分类器里边，它们结果差距才能大。

如果这两个东西看起来很不一样，就假如说有两只猫，一只猫特别瘦，黑黑的。一只是橘猫，特别的胖，大小也不一样。但这个在图片来说这差距是很大的。我们整个做feature extraction的时候是要把这两张图在最后变成一样，就在最后的时候变得相似。输入的时候不相似，但是经过特征提取其实要把它变相似。这样送入到了Softmax它才会产生分类的作用。

接下来讲了卷积神经网络的计算过程以及整个模型的搭建是什么样的。然后还讲了RES-NET的原理，这个也需要去理解。

Transfer Learning

那么现在，咱们今天就跟大家来介绍一个比较重要的概念，深度学习共同的基础部分，就是transfer learning。

咱们现在的这个深度学习模型变得越来越复杂了。上节课给大家举过这些例子，不同的人提出来了不同的模型，
重点给大家介绍了一个RES-NET和Inception model，也称为GoogleNET。

模型现在其实已经变得越来越复杂，这么复杂的结果是什么呢？结果是我们现在已经很难从头到尾搭建一个模型了。现在的模型结构已经这么复杂了，很少有人能有时间，或者在工作的时候有时间、有精力能从零开始一层一层的去做搭建，这是第一方面。

第二个方面，大家还发现一个特点。在结构中越接近前边虽然任务不一样，比如解决动物分类或者解决人物分类，但是越靠近前边，它们的特征相似度越高。

换句话说，有一个RES-NET专门对人物分类，还有一个是是专门做动物，它们分的类别完全不一样。但是就前边这些CNN的结果往往都很相似，而且是越往前越相似。

这是因为这些过程都是在做特征提取，如果都是一个比较相似的图片任务的话，在这个过程中特征提取其实从刚开始的时候在解析图片上的重要程度，其实要提取的东西都是类似的。

比方说识别我左手的水和我右手的手机，还有我前面站着一个美女，刚开始都是要识别它的轮廓。然后都要识别它的局部的形状，还要识别颜色… 这样的一个直接的结果，其实我们每一层用的filter都是类似的，只要达到一个比较好的结果，前面的这些filter都是类似的。

filter类似是因为filter控制的是我们要提取什么重要特征。那么我们就发现从前到后，其实越是前边越是比较简单的特征，线、块这些，到后边越来越综合。

有了这个之后大家就发现，既然现在模型这么复杂，从头到尾要搭建一个模型已经很难了，我们可以直接用这个模型的结构。

第二我们发现不仅模型的结构可以，模型的权重都也可以。可以用这个模型的权重来训练，直接把这个模型的权重拿过来。

其实也就是说，我们可以直接下载一个模型，把别人训练好的权重一起拿过来，这些东西就是一堆数字。然后它是在task a上弄的，我把它用到了task b上。训练的时候让它不要进行反向传播，在进入全连接层的时候再进行反向传播。

大家把这种学习方式就叫做transfer Learning， 迁移学习。我们平时日常在工作的时候经常会这么做。

客观上来讲，不同的任务，任务越类似肯定迁移的时候越好迁移。所以说其实它和任务的相似度以及和数据量的相对大小很有关系。

假设我们两个任务，A和B。这两个任务，A是分类狗，B是分类狼，A原本训练数据集是100W，B的训练集是1W。那么这两个任务比较而言，任务相似度非常大，原任务相对新任务数据量比较大，这个时候基本上迁移学习就非常好迁移，我们都可以不去更改进入全连接层之前的所有内容就可以进行迁移，只需要更改全连接层。也就是特征提取的部分完全平移。

那么如果A任务还是分类狗，B任务是分类汽车。A原本训练数据集是100W，B的训练集是5000W。那这两个任务比较而言，任务相似度非常小，原任务相对新任务数据量是小的，这个时候迁移学习就变得很困难，可能也只有图像线条，颜色这些个特征提取的部分可以迁移，基本是特征提取的最前边的部分。

所以，Transfer Learning的容易程度，在一个二维平面直角坐标系内的两个相关项，也就是x和y轴就是任务相似度和原任务相对新任务数据量的大小。

如果重新训练，怎么样来transfer呢？说了这么多，还是直接来看一个实例, 来看看我们具体该如何做「冻结」。

用的这个数据集, cifar10，这也是一个很经典的数据，它是十个典型的很常见的物品的分类。

咱们先引入必要的库，然后down数据集：

import torchvision
import torch.nn.functional as F
from torchvision.transforms import transforms
import torch
import matplotlib.pyplot as plt
from torch import nn
from icecream import ic


cifar_10 = torchvision.datasets.CIFAR10('.', download=True)

这个文件一共170多兆，大部分人物提取的特征差不多，所以权重可以不用更新，用其他相似任务的参数，相当于新模型初始化的时候，理解为更接近在最优点附近。

它里面的每一个数据的类型是一个PRL的image, 要在PyTorch里对这个图片进行使用，我们需要进行一个预处理。我们需要在前面定义一个方法：

preprocess = transforms.Compose([
    transforms.Resize(224),
    transforms.CenterCrop(224),
    transforms.ToTensor()
])

首先, 我们要先Resize，然后用一个CenterCrop， 让图片以中心扩散进行切割。如果有些图片不是正方形，那么第二个操作就是把中间的部分裁一个正方形出来。最后再把它变成一个Tensor。

然后我们需要修改一下数据获取数据时的transform。

cifar_10 = torchvision.datasets.CIFAR10('.', download=True, transform=preprocess)

现在看一下, cifar_10的数据就变成tensor了，shape是[3,224,224]

cifar_10[0][0].shape

---
torch.Size([3, 224, 224])

得到Tensor数据之后，要训练的时候得一次一次的取不同的数值出来，我们要做SGD，随机梯度下降。那么在做这个的时候有一种方法，写个复循环然后每次随机取一些index，再把这些index的值给它取出来，这是一种方法。

还有一种方法，我们可以直接用DataLoader，声明了之后每次要生成一个迭代器，每次会输出一些内容。

train_loader = torch.utils.data.DataLoader(cifar_10, batch_size=512, shuffle=True)

如果要把所有的数据传输进去，它有5万个照片太大了，内存吃不消。所以要把它做成SGD，要每次随机取一个东西。

然后我们来定义一个RES-NET：

resnet = torchvision.models.resnet18()

有了这样的RES-NET之后，它输出的是1000维的，而我们这里其实是需要一个10维的，那我们就需要把它的最后一层给它重新做一下。

feature_num = resnet.fc.in_features
resnet.fc = nn.Linear(feature_num, 10)

如果我们没有这一句，我们可以来看看它会输出什么。

ic(resnet(cifar_10[0][0].unsqueeze(0)))

---
ic| resnet(cifar_10[0][0].unsqueeze(0)): tensor([[-6.9484e-01, ..., 1008e+00]], grad_fn=<AddmmBackward0>)
tensor([[-6.9484e-01, ..., 1.1008e+00]],
       grad_fn=<AddmmBackward0>)

输出的是一个很长的东西，其实是有1,000维的，这里输出了1,000个。

现在如果把它的最后一层全连接层改了，变成10分类，因为这个cifar10是一个是分类问题。

feature_num = resnet.fc.in_features
resnet.fc = nn.Linear(feature_num, 10)
ic(resnet(cifar_10[0][0].unsqueeze(0)))

改完之后输出的数据就是10维的了，大家可以自己去跑一下代码，我这里就不贴了。

接着我们再来生成一个loss函数和一个优化器。

criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.SGD(resnet.parameters(), lr=1e-3, momentum=0.9)

criterion是测量尺度、考核标准的意思。parameters是要把所有参数进行拟合， 进行重新训练。

epochs = 2
losses = []

for epoch in range(epochs):
    epoch_loss = 0
    for i, (images, labels) in enumerate(train_loader):
        ic(epoch, i)
        predicts = resnet(images)
        loss = criterion(output, labels)
        optimizer.zero_grad()

        loss.backward()
        optimizer.step()

        epoch_loss += loss.item()

        if i > 0:
            print('Epoch: {} batch: {}, loss ==> {}'.format(epoch, i, epoch_loss / i))
    losses.append(epoch_loss / i)

plt.plot(losses)

---

0it [00:00, ?it/s]
ic| epoch: 0, i: 0
1it [00:53, 53.71s/it]ic| epoch: 0, i: 1
2it [01:42, 50.98s/it]
Epoch: 0 batch: 1, loss ==> 4.71190333366394
ic| epoch: 0, i: 2
...
98it [1:12:04, 44.13s/it]
Epoch: 1 batch: 97, loss ==> 1.7719330222336287

现在是这么个结果, 我们先来保存一下，我创建了一个32.log, 用于暂时保存咱们的结果。那因为我训练的时候加了一个tqdm，所以也把时间打印了出来，不过为了避免代码上的误解，所以代码我还是给的没有加tqdm的样子。

现在要迁移怎么迁移呢？很简单，第一步我们需要改一下我们的RES-NET。

resnet = torchvision.models.resnet18(pretrained=True)

我们加一个参数pretrained，然后将值设为True，现在要保留它的数据，保留之前训练的权重。

第二步要冻结它的这些参数，把RES-NET里边所有的parameters，每一个都有一个requires grad，给它定义成false。

for param in resnet.parameters():
    param.requires_grad = False # frozen weights

设置成false之后进行反向传播的时候这个值就不更新了。不更新的话那就相当于冻结了。

之前写的resnet.fc就相当于重写了fc分类层。

resnet.fc = nn.Linear(feature_num, 10) # rewrite fc classifier

假设现在的任务和原来任务不相似，或者说现在原来数据量和现在数据量相比偏小，那么对于这个RES-NET，不能把它所有的requires grad设置成false，要把它前面部分的给它设置成false，后边设置成true。

重写了这个FC classifire之后，新声明的参数默认它是需要进行梯度下降的，所以不需要在这写成false。就在这里，这个FC的grad默认是true。

那到这一步， transfer就结束了，我们可以重新训练来看看。

你会发现，时间上明显快多了。这个就是因为咱们这次训练的参数少了很多。

0it [00:00, ?it/s]ic| epoch: 0, i: 0
1it [00:17, 17.99s/it]ic| epoch: 0, i: 1
2it [00:34, 16.92s/it]
Epoch: 0 batch: 1, loss ==> 5.019284725189209
ic| epoch: 0, i: 2
3it [00:50, 16.54s/it]
Epoch: 0 batch: 2, loss ==> 3.7500953674316406
ic| epoch: 0, i: 3
...
98it [26:04, 15.96s/it]
Epoch: 1 batch: 97, loss ==> 1.108948134884392
...

之前我们每一轮训练几乎都要花个50s左右，现在基本在16左右，速度上提升了3倍。从总时间上我们也可以看出来，训练速度提升了好几倍，从原来的一小时12分钟，直接降到了26分钟。并且，loss也有所提升。

那么我们该怎么去看这个模型的层数，确定哪些是在前面部分，哪些实在后面呢？对于一个模型而言，最简单的办法就是直接print出来，比如说咱们的resnet18：

from torchvision.models import resnet18
print(resnet18())

---
ResNet(
  (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
  (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
  (relu): ReLU(inplace=True)
  (maxpool): MaxPool2d(kernel_size=3, stride=2, padding=1, dilation=1, ceil_mode=False)
  (layer1): Sequential(
    (0): BasicBlock(
      (conv1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    )
    (1): BasicBlock(
      (conv1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    )
  )
  (layer2): Sequential(
    (0): BasicBlock(
      (conv1): Conv2d(64, 128, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (downsample): Sequential(
        (0): Conv2d(64, 128, kernel_size=(1, 1), stride=(2, 2), bias=False)
        (1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      )
    )
    (1): BasicBlock(
      (conv1): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    )
  )
  (layer3): Sequential(
    (0): BasicBlock(
      (conv1): Conv2d(128, 256, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (downsample): Sequential(
        (0): Conv2d(128, 256, kernel_size=(1, 1), stride=(2, 2), bias=False)
        (1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      )
    )
    (1): BasicBlock(
      (conv1): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    )
  )
  (layer4): Sequential(
    (0): BasicBlock(
      (conv1): Conv2d(256, 512, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (downsample): Sequential(
        (0): Conv2d(256, 512, kernel_size=(1, 1), stride=(2, 2), bias=False)
        (1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      )
    )
    (1): BasicBlock(
      (conv1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (relu): ReLU(inplace=True)
      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    )
  )
  (avgpool): AdaptiveAvgPool2d(output_size=(1, 1))
  (fc): Linear(in_features=512, out_features=1000, bias=True)
)

这次我将结果打全，我们可以清晰的看到这个模型里从上到下，从前到后的每一层，最后一层是一个fc。

那除此之后，其实我们可以借用第三方库来进行计算，有一个库叫做torchsummary，

from torchsummary import summary
summary(resnet18(), (3, 224, 224))

---
----------------------------------------------------------------
        Layer (type)               Output Shape         Param #
================================================================
            Conv2d-1         [-1, 64, 112, 112]           9,408
            ...
AdaptiveAvgPool2d-67            [-1, 512, 1, 1]               0
           Linear-68                 [-1, 1000]         513,000
================================================================
Total params: 11,689,512
Trainable params: 11,689,512
Non-trainable params: 0
----------------------------------------------------------------
Input size (MB): 0.57
Forward/backward pass size (MB): 62.79
Params size (MB): 44.59
Estimated Total Size (MB): 107.96
----------------------------------------------------------------

这个去监测模型的层数和信息就更好一些，可以很直观的看到每一层以及整个模型的相关信息。不管是你自己的模型还是第三方预先训练好的其实都可以。我们在后面设置了一下输入的大小，设置了之后，summary在后面参数一共多少就一个一个都给你显示出来了。我们刚才输入的(3, 224, 224)，然后从第一层开始的Output Shape是多少，一层一层的向下就直接有了。

这两个方式都还是很有用的。

那么之后做训练的时候大家要对几个数字稍微多一点敏感性，我们来看，首先我们定义一个loss函数：

def cross_entropy(y, yhat): return -np.sum(y*np.log2(yhat))

然后我们输入下面几个值做测试：

cross_entropy([0, 1], [0.5]*2)
cross_entropy([0, 0, 0, 0, 1], [0.2]*5)
cross_entropy([0]*9 + [1], [0.1] * 10)
cross_entropy([0, 0, 1], [0.33] * 3)

---
1.0
2.321928094887362
3.321928094887362
1.5994620704162712

transfer learning基于的是模型从前往后。前面层学的东西比较基础，到后边学的抽象层次越来越高，看到的是更复杂的一些。

那咱们现在就再来演示一下它到底学的学到都是什么东西。那么为了看一下这个到底学的是什么，我再次贡献一下自己。

这个是早些时候我一个同学帮我画的头像，就拿它来看吧。

首先，我们前面看到打印结果了，resnet18的第一层是conv1，我们来看看第一层：

from PIL import Image

preprocess = transforms.Compose([
    transforms.Resize(224),
    transforms.CenterCrop(224),
    transforms.ToTensor()
])
resnet = torchvision.models.resnet18(pretrained=True)

myself = preprocess(Image.open('./assets/chaheng2.png'))
resnet.conv1(myself.unsqueeze(0))

然后我们就可以看到一堆的tensor数据，这个unsqueeze是将数据改变了一下结构，从myself变成了[[myself]]，改成这样是因为torch每次接收的是一个batch的东西，直接输入一个图片是不行的。

我们看一下它的这个输出,第一个卷积的输出是什么：

我们可以看到，它的shape是[1, 64, 112, 112],那这里边的分别是什么？

第一个维度，这个1是batch的数量。64是filter的channel，所以它输出了64张图片。 后面的112和112是一组数据，从这个数据来看，这个图片经过卷积之后，经历了一个缩小的变化。从原来的224缩小到了112， 经历了一个下采样。

接着咱们来看一下具体的数据内容，看看output第0个的内容是什么样的：

plt.imshow(output[0][0].detach())

因为结果还在内存里，所以我们永乐一个detach()。

它把我的轮廓给提出来了。

我们再来看看别的是什么样，我们改成[0][2]：

这个貌似是将背景扣了。

我们不一张一张来看了，咱们来将探索过程写个循环，看一下它到底都做了什么。

width = 8
fig, ax = plt.subplots(output[0].shape[0] // width, width, figsize=(20, 20))

for i in range(output[0].shape[0]):
    ix = np.unravel_index(i, ax.shape)
    plt.sca(ax[ix])
    ax[ix].title.set_text('filter-{}'.format(i))

    plt.imshow(output[0][i].detach())

plt.show()

它这64个学到的几乎每个都不一样，那有些是有用的，有些是没用的。有些是从边缘层面上，比如说filter20就是从边缘上，而有一些，比如filter21就是从颜色上。

那么如果我们现在想把第二个、第三个、第四个这些都拿出来的话怎么办？当然理论上可以沿着它的结构给一层一层解出来，但是PyTorch里面给咱们的提供了一个比较简单的方法。

那刚才写的那个代码，其实是在进行前向传播，就我们刚才写代码就是在模拟它的前向传播，forward。PyTorch就给我们提供了一个很方便东西，它可以给前向传播及反向传播的时候注册一个函数。就比如说：

conv_model = [m for _, m in resnet.named_modules() if isinstance(m, torch.nn.Conv2d)]

for m in conv_model:
    m.register_forward_hook()

我们现在把resnet里边所有的model拿出来，然后如果这个model它是卷机，给这些所有的模型注册一个函数。这个函数是是他在进行前向传播的时候会自己调用的，就不需要咱们再手动的去写了。

那我们现在就来将之前写的内容抽象成一个函数visualize_model，在定义这个函数的时候需要注意一下PyTorch的相关API，

那我们在定义的时候，也就需要一样传递这些参数：

def visualize_model(model, input_, output):
    width = 8
    ...
    plt.show()

这样前向传播的时候,它会自动调用。现在我们就可以让它来进行前向传播：

for m in conv_model:
    m.register_forward_hook(visualize_model)

with torch.no_grad():
    resnet(myself.unsqueeze(0))

no_grad的意思是不让它进行反向传播，只进行前向传播。

我们在观察它每一层的结果的时候，就会发现越到后面就越抽象，我们捡中间某一张贴出来来看。

就基本上，这个时候还能勉强辨认出是个任务头像，再往后的结果，我肉眼已经分辨不出来它是个啥了。就这是整个模型一层一层学到的东西，它里边是从刚开始的时候比较的底层、比较的基础，后边会提取的东西越来越多。

就咱们在这里所做的这种权重可视化，有一个比较有趣的应用：deep dream，它就是将刚才这些学到的靠后的权重，然后应用到一张图片上。就我们刚刚可视化那种层数再应用到一些新图片上，就会产生这样的效果。如果感兴趣的可以自己试一下。

好那咱们这个RES-NET和RES-NET可视化，以及transfer learning的内容，到这里就可以告一段落了。整个的深度学习的基础部分，也就到这里结束了。

最后，我们来留一个小作业。

作业

那么本节课的最后，给大家留一个小作业，稍微还是有点难度的，需要大家自己去查阅相关手册才行，不过知识点都是讲过的。作业内容为**「对验证码进行识别」**。

1. 练习内容： 训练一个模型，对验证码中的字符进行分类识别，并最终完成验证码识别的任务。

2. 数据集： 数据集内包含0-9以及A-Z一共36个字符，训练集中每个字符有50张图片，验证集中每个字符有10张图片，验证码数据集是由随机去除的4个字符图片拼接而成。

3. 需要的相关知识：

• 数据读取
• 使用torch搭建、训练、验证模型
• 模型预测于图片切分

好，给大家提供下思路，我们将我们需要解决的问题分成四步：第一个，先建立字符对照表，第二个，要定义一个datasets和一个dataloader。 第三个，需要定义网络结构。 第四个，定义模型训练函数。 最后，就是验证训练结果。

数据集如下：

链接: https://pan.baidu.com/s/1Ua-nJGqkQO_nDZaLyFqE7g?pwd=uvsd 提取码: uvsd
–来自百度网盘超级会员v6的分享