目标给模型输入一张纯色的图片模型识别并输出其颜色一、数据准备这里我们将使用经典的 CIFAR10 数据集它包含 10 个类别的彩色图像每个类别有 6000 张图像图像大小为 32x32 像素。①使用 torchvision 加载 CIFAR10 数据集import torch import torchvision import torchvision.transforms as transforms ## 数据预处理将图像转换为张量并进行标准化 transform transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) ## 下载训练集和测试集 trainset torchvision.datasets.CIFAR10(root./data, trainTrue, downloadTrue, transformtransform) trainloader torch.utils.data.DataLoader(trainset, batch_size4, shuffleTrue, num_workers2) testset torchvision.datasets.CIFAR10(root./data, trainFalse, downloadTrue, transformtransform) testloader torch.utils.data.DataLoader(testset, batch_size4, shuffleFalse, num_workers2) ## 定义类别名称 classes (plane, car, bird, cat, deer, dog, frog, horse, ship, truck)1数据预处理作用通过一系列操作将图像数据转换为适合神经网络输入的形式。transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])组合多个图像变换操作2数据集下载与加载训练集与测试集划分CIFAR-10 数据集被划分为训练集和测试集训练集用于训练模型测试集用于评估模型性能。torchvision.datasets.CIFAR10用于加载 CIFAR-10 数据集。root./data指定数据集的存储路径若数据集不存在则会下载到该目录。trainTrue表示加载训练集。downloadTrue若数据集不存在则自动下载。transformtransform对数据集应用之前定义的变换操作。torch.utils.data.DataLoader用于创建数据加载器将数据集打包成小批量数据方便模型训练。trainset和testset分别是训练集和测试集。batch_size4每次迭代加载 4 个样本。shuffleTrue在训练集中打乱数据顺序使模型在训练时看到更多样的数据分布。shuffleFalse在测试集中不打乱数据顺序便于评估模型性能。num_workers2指定使用 2 个子进程加载数据可加快数据加载速度。3类别名称定义classes (plane, car, bird, cat, deer, dog, frog, horse, ship, truck)定义了 CIFAR-10 数据集中 10 个类别的名称用于后续对预测结果的解释和分析。②数据可视化import matplotlib.pyplot as plt import numpy as np ## 定义一个函数用于显示图像 def imshow(img): img img / 2 0.5 # 反标准化 npimg img.numpy() plt.imshow(np.transpose(npimg, (1, 2, 0))) plt.show() ## 获取一批训练数据 dataiter iter(trainloader) images, labels next(dataiter) ## 显示图像 imshow(torchvision.utils.make_grid(images)) ## 打印标签 print( .join(f{classes[labels[j]]:5s} for j in range(4)))二、定义卷积神经网络模型颜色分类器核心①定义 CNN 架构熟悉流程以及神经网络细节可见往期文章点击链接即可跳转import torch.nn as nn import torch.nn.functional as F class Net(nn.Module): def __init__(self): super(Net, self).__init__() # 定义卷积层和池化层 self.conv1 nn.Conv2d(3, 6, 5) # 输入通道3输出通道6卷积核大小5 self.pool nn.MaxPool2d(2, 2) # 最大池化层窗口大小2步长2 self.conv2 nn.Conv2d(6, 16, 5) # 输入通道6输出通道16卷积核大小5 # 定义全连接层 self.fc1 nn.Linear(16 * 5 * 5, 120) self.fc2 nn.Linear(120, 84) self.fc3 nn.Linear(84, 10) def forward(self, x): # 前向传播过程 x self.pool(F.relu(self.conv1(x))) # 卷积 激活 池化 x self.pool(F.relu(self.conv2(x))) # 卷积 激活 池化 x x.view(-1, 16 * 5 * 5) # 展平操作 x F.relu(self.fc1(x)) # 全连接 激活 x F.relu(self.fc2(x)) # 全连接 激活 x self.fc3(x) # 输出层 return x net Net() print(net)卷积层self.conv1 nn.Conv2d(3, 6, 5)第一个卷积层输入通道数为 3对应 RGB 三通道图像输出通道数为 6卷积核大小为 5。该层用于提取图像的局部特征。self.pool nn.MaxPool2d(2, 2)最大池化层窗口大小为 2步长为 2用于降低特征图的空间维度减少计算量同时保留重要特征。self.conv2 nn.Conv2d(6, 16, 5)第二个卷积层输入通道数为 6输出通道数为 16卷积核大小为 5。进一步提取更抽象的特征。全连接层self.fc1 nn.Linear(16 * 5 * 5, 120)第一个全连接层输入大小为 16 * 5 * 5输出大小为 120。将卷积层提取的特征映射到更高维的空间进行更复杂的数据表示。self.fc2 nn.Linear(120, 84)第二个全连接层输入大小为 120输出大小为 84。继续对特征进行变换和整合。self.fc3 nn.Linear(84, 10)输出层输入大小为 84输出大小为 10对应 CIFAR - 10 数据集的 10 个类别。前向传播在forward方法中输入图像先经过第一个卷积层应用ReLU激活函数后进行最大池化然后重复该过程经过第二个卷积层。之后将特征图展平并依次通过三个全连接层前两个全连接层后都应用ReLU激活函数最后一层直接输出分类结果。三、定义损失函数和优化器模型训练的指引①损失函数 优化器损失函数用于衡量模型预测结果与真实标签之间的差距优化器则负责根据损失函数的梯度信息更新模型参数。损失函数优化器详解见往期文章import torch.optim as optim # 使用交叉熵损失函数和随机梯度下降优化器 criterion nn.CrossEntropyLoss() optimizer optim.SGD(net.parameters(), lr0.001, momentum0.9)criterion nn.CrossEntropyLoss()定义损失函数 使用交叉熵损失函数衡量模型输出与真实标签之间的差异。optimizer optim.SGD(net.parameters(), lr0.001, momentum0.9)定义优化器 使用随机梯度下降SGD优化器lr0.001表示学习率控制参数更新步长momentum0.9表示动量因子加速梯度下降过程。tips交叉熵损失函数是一种用于衡量两个概率分布之间差异的函数主要用于分类问题中的损失计算。详细解释可见大佬文章四、训练网络提升模型性能训练过程是模型学习数据特征、优化参数的关键环节。我们需要多次迭代训练数据逐步调整模型参数以降低损失函数的值。for epoch in range(2): # 遍历数据集多次 running_loss 0.0 for i, data in enumerate(trainloader, 0): # 获取输入数据和标签 inputs, labels data # 清空梯度缓存 optimizer.zero_grad() # 前向传播 反向传播 优化 outputs net(inputs) loss criterion(outputs, labels) loss.backward() optimizer.step() # 打印统计信息 running_loss loss.item() if i % 2000 1999: # 每 2000 个小批量打印一次 print(f[{epoch 1}, {i 1:5d}] loss: {running_loss / 2000:.3f}) running_loss 0.0 print(Finished Training)这段代码是训练神经网络的核心部分它实现了模型的训练过程在每个训练周期epoch中依次处理训练集中的每个小批量数据通过前向传播计算输出和损失再通过反向传播更新模型参数。遍历数据集多次-初始化损失统计变量-遍历每个小批量数据-获取输入数据和标签-清空梯度缓存-前向传播 反向传播 优化-打印统计信息打印统计信息running_loss loss.item()将当前小批量的损失值累加到running_loss中。if i % 2000 1999: # 每 2000 个小批量打印一次每处理 2000 个小批量后打印一次平均损失信息。print(f[{epoch 1}, {i 1:5d}] loss: {running_loss / 2000:.3f})打印 当前训练周期和小批量索引以及过去 2000 个小批量的平均损失值。running_loss 0.0重置running_loss以便开始累计下一个 2000 小批量的损失值。tips:如果电脑配备 GPU可以利用 GPU 加速模型训练过程显著提升训练速度。device torch.device(cuda:0 if torch.cuda.is_available() else cpu) net.to(device) ## 将输入数据和标签移动到 GPU 上 inputs, labels inputs.to(device), labels.to(device)五、保存和加载模型模型持久化与复用训练完成后我们可以将模型参数保存到文件中以便后续加载和使用。PATH ./cifar_net.pth torch.save(net.state_dict(), PATH)加载模型参数net Net() net.load_state_dict(torch.load(PATH))六、测试网络评估模型性能correct 0 total 0 with torch.no_grad(): # 在测试阶段不需要计算梯度 for data in testloader: images, labels data outputs net(images) _, predicted torch.max(outputs.data, 1) total labels.size(0) correct (predicted labels).sum().item() print(fAccuracy of the network on the 10000 test images: {100 * correct / total:.2f}%)禁用梯度计算在测试阶段模型参数已经训练完成不需要再计算梯度或更新参数。使用torch.no_grad()可以临时关闭梯度计算节省内存并加快计算速度。遍历测试数据遍历测试集testloader中的每个小批量数据。每个小批量数据包含images输入图像数据。labels图像对应的真实标签。模型预测outputs net(images)将输入图像images通过训练好的模型net进行前向传播得到模型的输出outputs。torch.max(outputs.data, 1)对模型输出进行处理返回每个样本预测概率最大的类别索引_表示最大概率值我们不关心。predicted表示模型预测的类别索引。统计预测结果total labels.size(0)累计当前小批量的样本数量到总样本数total中。(predicted labels)比较预测类别与真实标签返回一个布尔张量。.sum().item()统计预测正确的样本数量并转换为 Python 数字累加到correct中。打印准确率总结本文介绍了PyTorch训练图像分类器的核心步骤包括数据准备、网络定义、模型训练、性能评估以及GPU 加速等关键技术。引用文献资料PyTorch 神经网络_w3cschoolhttps://www.w3cschool.cn/pytorch/pytorch-w18e3be1.html交叉熵损失函数解析-CSDN博客https://blog.csdn.net/b1055077005/article/details/100152102?ops_request_miscelastic_search_miscrequest_id32f6dd1f5f7a13d1801a3dbf24560931biz_id0utm_mediumdistribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-100152102-null-null.142^v102^pc_search_result_base3utm_term%E4%BA%A4%E5%8F%89%E7%86%B5%E6%8D%9F%E5%A4%B1%E5%87%BD%E6%95%B0spm1018.2226.3001.4187