目录

• 1 数据集Dataset
• 2 数据加载DataLoader
• 3 常用预处理方法
• 4 模型处理
• 5 实例：MNIST数据集处理

1 数据集Dataset

Dataset类是Pytorch中图像数据集操作的核心类，Pytorch中所有数据集加载类都继承自Dataset父类。当我们自定义数据集处理时，必须实现Dataset类中的三个接口：

• 初始化

  def __init__(self)
  

  构造函数，定义一些数据集的公有属性，如数据集下载地址、名称等
• 数据集大小

  def __len__(self)
  

  返回数据集大小，不同的数据集有不同的衡量数据量的方式
• 数据集索引

  def __getitem__(self, index):
  

  支持数据集索引功能，以实现形如dataset[i]得到数据集中的第i + 1个数据的功能。__getitem__是后期迭代数据时执行的具体函数，其返回值决定了循环变量，例如

  class data(Dataset)...def __getitem__(self, idx: int):if self.transforms:img = self.transforms(img)return img, label			# 返回的值即为后续迭代的循环变量for images, labels in dataLoader:...
  

2 数据加载DataLoader

为什么有了数据集Dataset还需要数据加载器DataLoader呢？原因在于神经网络需要进一步借助DataLoader对数据进行划分，也就是我们常说的batch，此外DataLoader还实现了打乱数据集、多线程等操作。

DataLoader本质是一个可迭代对象，可以使用形如

for inputs, labels in dataloaders

进行可迭代对象的访问。

我们一般不需要去实现DataLoader的接口，只需要在构造函数中指定相应的参数即可，比如常见的batch_size，shuffle等参数。

下面这张图非常好地说明了Dataset和DataLoader的关系

接下来总结数据构造的三步法

1. 继承Dataset对象，并实现__len__()、__getitem__()魔法方法，该步骤的主要目的在于将文件形式的数据集处理为模型可用的标准数据格式，并加载到内存中；
2. 用DataLoader对象封装Dataset，使其成为可迭代对象；
3. 遍历DataLoader对象以将数据加载到模型中进行训练。

3 常用预处理方法

在数据集Dataset的__getitem__()中利用torchvision.transforms进行数据预处理与变换

常见的数据预处理变换方法总结如下表

4 模型处理

考虑以下场景：

网络的部分层级结构已经收敛、无需调整；大型复杂网络需要微调(Fine-tune)某些结构或参数；希望基于已训练好的模型进行改善或其他研究工作。

这些场景下重新通过数据集训练整个神经网络并无必要，甚至会使模型不稳定，因此引入预训练(pretrained)。Pytorch允许用户保存已训练好的模型，或加载其他模型，避免往复的无谓重训练，其中模型参数文件以.pth为后缀

# 保存已训练模型
torch.save(model.state_dict(), path)
# 加载预训练模型
model.load_state_dict(torch.load(path), device)

通过设置模型某些层可学习参数的requires_grad属性为False即可固定这部分参数不被后续学习过程影响。深度学习框架应用优势之一在于预设了对GPU的支持，大大提高模型处理与训练的效率。Pytorch中通过mode.to(device)方法将模型部署到指定设备上(CPU/GPU)，范式如下：

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model.to(device)

工程上也常使用torch.nn.DataParallel(model, devices)来处理多GPU并行运算，其原理是：首先将模型加载到主GPU上，再将模型从主GPU产生若干副本到其余GPU，随后将一个batch中的数据按维度划分为不同的子任务给各GPU进行前向传播，得到的损失会被累积到主GPU上并由主GPU反向传播更新参数，最后将更新参数拷贝到其余GPU以开始下一轮训练。

5 实例：MNIST数据集处理

下面给出了处理MNIST手写数据集的完整代码，可以用于加深对数据处理流程的理解

from abc import abstractmethod
import numpy as np
from torchvision.datasets import mnist
from torch.utils.data import Dataset
from PIL import Imageclass mnistData(Dataset):'''* @breif: MNIST数据集抽象接口* @param[in]: dataPath -> 数据集存放路径* @param[in]: transforms -> 数据集变换'''    def __init__(self, dataPath: str, transforms=None) -> None:super().__init__()self.dataPath = dataPathself.transforms = transformsself.data, self.label = [], []def __len__(self) -> int:return len(self.label)def __getitem__(self, idx: int):img = self.data[idx]if self.transforms:img = self.transforms(img)return img, self.label[idx]@abstractmethoddef plot(self, index: int) -> None:pass@abstractmethoddef load(self) -> list:passdef plotData(self, index: int, info: str=None) -> None:'''* @breif: 可视化训练数据* @param[in]: index -> 数据集索引* @param[in]: info -> 备注信息* @retval: None'''print(info, " --index:", index, "--label:", self.label[index])  if info else \print(" --index:", index, "--label:", self.label[index])          img = Image.fromarray(np.uint8(self.data[index]))img.show()def loadData(self, train: bool) -> list:'''* @breif: 下载与加载数据集* @param[in]: train -> 是否为训练集* @retval: 数据与标签列表'''    # 如果指定目录下不存在数据集则下载dataSet   = mnist.MNIST(self.dataPath, train=train, download=True)# 初始化数据与标签data  = [ i[0] for i in dataSet ]label = [ i[1] for i in dataSet ]return data, labelclass mnistTrainData(mnistData):'''* @breif: MNIST训练集* @param[in]: dataPath -> 数据集存放路径* @param[in]: transforms -> 数据集变换'''    def __init__(self, dataPath: str, transforms=None) -> None:super().__init__(dataPath, transforms=transforms)self.data, self.label = self.load()def plot(self, index: int) -> None:self.plotData(index, "trainSet data")def load(self) -> list:return self.loadData(train=True)class mnistTestData(mnistData):'''* @breif: MNIST测试集* @param[in]: dataPath -> 数据集存放路径* @param[in]: transforms -> 数据集变换'''    def __init__(self, dataPath: str, transforms=None) -> None:super().__init__(dataPath, transforms=transforms)self.data, self.label = self.load()def plot(self, index: int) -> None:self.plotData(index, "testSet data")def load(self) -> list:return self.loadData(train=False)