PyTorch 实现 CIFAR10 图像分类知识点总结

一、数据加载与预处理

1. 工具库依赖：使用torchvision加载数据集，torchvision.transforms做数据变换，torch.utils.data.DataLoader实现批量数据加载。
2. 数据变换流程：
   • transforms.ToTensor()：将图像转为 PyTorch 张量，并把像素值归一化到[0,1]。
   • transforms.Normalize(mean, std)：对张量标准化（如 CIFAR10 用(0.5, 0.5, 0.5)作为均值和标准差），使像素值分布到[-1,1]，加速模型收敛。
3. 数据集加载：
   • 调用torchvision.datasets.CIFAR10(root, train, download, transform)，指定数据存储路径、训练 / 测试模式、是否自动下载、数据变换规则。
4. 数据加载器配置：
   • 通过torch.utils.data.DataLoader(dataset, batch_size, shuffle, num_workers)创建批量加载器，设置批次大小（如batch_size=4）、是否打乱数据（训练集shuffle=True，测试集shuffle=False）、工作线程数，提升数据迭代效率。

二、卷积神经网络（CNN）构建

1. 网络继承与结构：自定义网络类继承torch.nn.Module（如class CNNNet(nn.Module)），通过__init__定义层组件，forward定义数据流动逻辑。
2. 核心层组件：
   • 卷积层：nn.Conv2d(in_channels, out_channels, kernel_size, stride)，负责提取图像局部特征（如输入 3 通道、输出 16 通道、5×5 卷积核）。
   • 池化层：nn.MaxPool2d(kernel_size, stride)，对特征图下采样，减少参数与计算量，保留关键特征（如 2×2 池化核）。
   • 全连接层：nn.Linear(in_features, out_features)，将卷积特征映射到类别空间（如 CIFAR10 有 10 类，最终全连接层输出为 10）。
3. 前向传播逻辑：
   • 结合激活函数（如F.relu）、池化操作，以及张量变形（view）—— 将卷积输出的多维特征展平为全连接层的输入（如x = x.view(-1, 36*6*6)）。
4. 设备兼容性：通过torch.device("cuda:0" if torch.cuda.is_available() else "cpu")判断 GPU 是否可用，再用net.to(device)将模型移到对应设备（GPU/CPU）。

三、模型训练

1. 损失与优化配置：
   • 损失函数：选用nn.CrossEntropyLoss()，适用于多分类任务（内置 Softmax+NLLLoss，直接计算预测与真实标签的损失）。
   • 优化器：如optim.SGD(net.parameters(), lr=0.001, momentum=0.9)（带动量的随机梯度下降，加速收敛），或optim.Adam（自适应学习率，更灵活）。
2. 训练循环逻辑：
   • 多轮迭代（epoch）：遍历训练集多次（如range(10)表示训练 10 轮），提升模型泛化能力。
   • 批次迭代：每批数据执行以下步骤：
     • 数据上设备：inputs, labels = inputs.to(device), labels.to(device)。
     • 梯度清零：optimizer.zero_grad()（避免梯度累积影响参数更新）。
     • 前向传播：outputs = net(inputs)获取模型预测。
     • 损失计算：loss = criterion(outputs, labels)。
     • 反向传播：loss.backward()计算参数梯度。
     • 参数更新：optimizer.step()根据梯度更新模型参数。
   • 损失监控：定期打印批次损失（如每 2000 批打印一次），观察训练趋势。

四、模型评估

1. 测试数据加载：用DataLoader加载测试集（shuffle=False，保证结果可复现）。
2. 预测与验证：
   • 前向传播：outputs = net(images)得到类别得分。
   • 提取预测类别：_, predicted = torch.max(outputs, 1)（torch.max返回 “最大值 + 对应索引”，索引即预测类别）。
   • 结果对比：将predicted与真实标签labels比较，评估分类效果（如查看单批样例的预测与真实值是否一致）。

五、辅助操作

• 图像可视化：结合matplotlib.pyplot和torchvision.utils.make_grid，将批量图像拼接后显示，直观查看数据或预测结果。
• 模型复杂度统计：通过sum(x.numel() for x in net.parameters())计算模型总参数数量，量化模型复杂度。

上述内容覆盖了数据处理、模型构建、训练优化、评估验证全流程，体现了 PyTorch 实现图像分类任务的典型思路与关键技术。