尧图网站建设 
简介这套基于深度学习LeNet-5模型的图像分类系统源码面向计算机相关专业的本科生毕业设计与课程设计人群适合需要快速搭建图像识别项目、理解经典卷积神经网络训练流程的学习者。项目用Python实现完整流程包含数据清洗与预处理、模型定义和训练、测试评估等模块最终准确率99.12%可作为课程报告或毕设演示的可靠蓝本。资源共27个文件压缩包约11.53MB。其中10个.py源码文件覆盖数据读取、模型构建、训练测试及辅助分析等功能2个.pth文件为训练好的模型参数便于直接加载复现4个.gz为MNIST原始数据集另有文本说明、项目配置等文件整体结构清晰便于按需查看。目前已有102人学习下载适合希望系统参考LeNet-5实际工程实现的初学者。读者可从中获得一套可运行的图像分类项目包括模型结构定义、梯度训练流程、数据增强与预处理技巧、测试脚本及使用说明。通过阅读源码还能掌握深度学习框架下从数据准备到模型部署的常见写法为后续改进或扩展其他分类任务打下基础。1. 从一份带准确率的项目压缩包说起LeNet-5为什么到今天仍值得跑第一次看到“基于深度学习LeNet-5模型的图像分类系统python源码准确率99.12%.zip”这样的标题大多数人的第一反应可能是“2026年了还有人用1998年的网络做分类”。这个反应本身没有错但恰恰忽略了一个事实LeNet-5在MNIST这类灰度小图像任务上配合正确的训练策略依然能以极小的参数量获得99%以上的准确率。换句话说这份压缩包卖的不是“新”而是“经典结构正确的训练方法”的组合。这个标题实际指向一个很具体的需求用Python实现一个能直接运行的图像分类系统输入手写数字图片输出类别并且准确率要能够对标论文级别的99.12%。它解决的典型场景包括课程设计、算法验证、简历项目以及刚入门深度学习时“第一个能跑通并看到高准确率”的正反馈。适合的人群是熟悉Python基础、但还没有完整训练过一个CNN模型的开发者。接下来要做的是从网络结构、数据处理、训练策略到推理部署把这个“99.12%”拆成一串可复现的参数和代码。2. LeNet-5网络结构拆解从C1到F6的参数表与PyTorch实现2.1 经典结构在MNIST上的两种改法LeNet-5原论文面向的是32×32灰度输入结构依次为卷积层C1、池化层S2、卷积层C3、池化层S4、卷积层C5、全连接层F6和输出层。原始C5是一个卷积分支输出的是一维特征向量在MNIST的28×28输入下很多实现会直接把C5改写成全连接层或者给第一层卷积加padding。两种改法都不影响“LeNet-5”这个名称的成立但直接影响后续全连接层的维度计算。第一种改法输入28×28第一层卷积不加padding5×5卷积核输出24×24池化后12×12第二种改法第一层卷积加padding2输出28×28池化后14×14。这两种计算路径在PyTorch中都很常见区别只在于你对原始架构还原到什么程度。下面采用padding2的方案因为它能让网络在MNIST上保留更多空间信息收敛更快这也是许多高准确率复现的默认选择。2.2 每一层在做什么参数表与计算逻辑以“1个灰度通道输入 → 6个5×5卷积核 → 平均池化 → 16个5×5卷积核 → 平均池化 → 展平 → 120 → 84 → 10”为骨干各层输出尺寸与参数量如下表层名操作卷积核/池化输出尺寸参数量C1卷积5×5, 6个28×28×6156S2平均池化2×2, stride 214×14×60C3卷积5×5, 16个10×10×162416S4平均池化2×2, stride 25×5×160C5全连接400→12012048120F6全连接120→848410164输出全连接84→1010850注意C3的参数量是2416而不是简单的16×25。原因是C3的每个卷积核都作用在S2的全部6个通道上所以单个卷积核的权重是6×5×5150加上1个偏置再乘以16个卷积核得到(1501)×162416。这个细节在阅读源码时容易困惑也是调试网络结构时最常被忽视的地方。2.3 用PyTorch实现一个可训练的LeNet-5下面这段代码定义了完整的网络输入为1通道灰度图输出为10类得分import torch.nn as nn class LeNet5(nn.Module): def __init__(self, num_classes10): super().__init__() self.features nn.Sequential( # C1: 1通道 - 6通道, padding2 保持28x28 nn.Conv2d(1, 6, kernel_size5, padding2), nn.ReLU(), # S2: 平均池化, 输出14x14 nn.AvgPool2d(kernel_size2, stride2), # C3: 6通道 - 16通道, 输出10x10 nn.Conv2d(6, 16, kernel_size5), nn.ReLU(), # S4: 输出5x5 nn.AvgPool2d(kernel_size2, stride2), ) self.classifier nn.Sequential( nn.Flatten(), # C5与F6: 16*5*5400 - 120 - 84 nn.Linear(16 * 5 * 5, 120), nn.ReLU(), nn.Linear(120, 84), nn.ReLU(), nn.Linear(84, num_classes) ) def forward(self, x): return self.classifier(self.features(x))代码逻辑说明features负责从原始像素中提取局部纹理和边缘组合classifier负责把空间特征映射到类别概率。nn.AvgPool2d是LeNet原始使用的平均池化与现在图像分类中更主流的nn.MaxPool2d不同平均池化对微小位移更平滑在MNIST上两者差距很小但早期复现项目里多为平均池化。参数说明padding2是保证28×28输入下C1输出仍然为28×28的关键去掉它会让后续nn.Linear(16 * 5 * 5, 120)直接报维度错误。如果你希望严格还原32×32输入可以把网络第一层去掉padding但需要把输入图像resize到32×32同时修改全连接层输入维度为16×5×5400在32×32输入加padding0时C3输出尺寸会不同务必用下面的验证方法确认。2.4 一个必须养成的习惯实例化后立即验证输出形状不要凭直觉去猜维度用一行代码确认网络输出import torch net LeNet5() dummy torch.randn(1, 1, 28, 28) print(net(dummy).shape) # torch.Size([1, 10])这一步能过滤掉绝大多数传入尺寸错误。如果输出不是[1, 10]优先检查卷积padding和全连接层输入维度是否与特征图展平后的长度一致。3. MNIST数据加载、标准化与训练循环跑通99%准确率的最小代码3.1 数据集的获取与预处理为什么要用这两个标准化参数MNIST是LeNet-5最经典的实验场PyTorch的torchvision.datasets.MNIST可以直接下载。训练集6万张、测试集1万张每张是28×28灰度图。预处理只有两步转成Tensor、标准化。import torch from torch.utils.data import DataLoader from torchvision import datasets, transforms transform transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,)) ]) train_dataset datasets.MNIST(root./data, trainTrue, downloadTrue, transformtransform) test_dataset datasets.MNIST(root./data, trainFalse, downloadTrue, transformtransform) train_loader DataLoader(train_dataset, batch_size64, shuffleTrue) test_loader DataLoader(test_dataset, batch_size1000, shuffleFalse)逻辑说明归一化让像素值从[0,1]映射到近似标准正态分布可以加速收敛。0.1307是MNIST训练集的全局均值0.3081是全局标准差这两个数值是torchvision文档和各类复现项目长期验证的固定值不必自己重算。参数说明shuffleTrue只用于训练集它打断样本顺序避免模型学到数据排列中的假规律测试集不需要打乱。batch_size64对于LeNet-5来说是常规选择显存占用很低CPU上也能流畅训练如果你用的是GPU可以提高到128或256准确率不会受到明显影响。3.2 训练循环SGD与交叉熵的组合下面是一个精简但完整的训练流程保留了进度输出与模型保存import torch.optim as optim import torch.nn as nn device torch.device(cuda if torch.cuda.is_available() else cpu) model LeNet5().to(device) criterion nn.CrossEntropyLoss() optimizer optim.SGD(model.parameters(), lr0.01, momentum0.9) def train_one_epoch(epoch): model.train() total_loss 0 for images, labels in train_loader: images, labels images.to(device), labels.to(device) optimizer.zero_grad() outputs model(images) loss criterion(outputs, labels) loss.backward() optimizer.step() total_loss loss.item() avg_loss total_loss / len(train_loader) print(fEpoch {epoch} | Loss: {avg_loss:.4f}) return avg_loss for epoch in range(1, 11): train_one_epoch(epoch) torch.save(model.state_dict(), flenet5_epoch{epoch}.pth)代码逻辑说明model.train()开启训练模式optimizer.zero_grad()清空梯度loss.backward()反向传播optimizer.step()更新权重。顺序不能颠倒漏掉zero_grad()会导致梯度累积表现为损失曲线抖动甚至发散。参数说明lr0.01配合momentum0.9是MNIST上最稳妥的起点。动量项让参数更新在梯度方向变化较小时保持惯性能有效穿过局部极小值。对比AdamSGD在这个任务上更容易稳定收敛到99%以上的区域但如果你用的是更深的网络或非MNIST数据集Adam可能更好调。损失函数用CrossEntropyLoss它内部已经包含Softmax计算不需要在网络输出层再手动加Softmax。3.3 训练参数速查表下表是复现99%以上准确率的一组推荐参数直接照抄即可起步。参数推荐值说明优化器SGDmomentum0.9比Adam更贴合经典复现学习率0.01过大容易震荡过小收敛慢Batch Size64CPU友好稳定性和速度的折中Epoch10第5轮左右可到99%10轮彻底收敛数据预处理Mean0.1307, Std0.3081固定值不要自行计算池化方式AvgPool2d还原LeNet原始设计也可替换为MaxPool2d如果训练到第10轮准确率卡在98%左右大概率不是网络代码问题而是随机种子、数据顺序或学习率策略的影响。这正是下面一章要展开的内容。4. 从97%到99.12%LeNet-5的训练策略、超参与准确率验证4.1 为什么同样的网络有人99.12%有人98.3%随机种子、shuffle顺序、学习率是否衰减、池化层选平均还是最大这些因素叠加在一起会在MNIST上产生0.5%到1%的准确率波动。99.12%并不是一个靠运气才能碰到的天花板而是几项关键策略同时起作用的自然结果。第一项策略是固定随机种子。PyTorch中数据加载与权重初始化都依赖随机数不固定种子的话每次运行结果会有细微差异。在训练脚本开头加入import random import numpy as np import torch def set_seed(seed42): random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed_all(seed) set_seed(42)逻辑说明torch.manual_seed控制权重初始化和CUDA相关随机数random.seed与np.random.seed影响数据集shuffle。三者缺一不可只固定torch而忽略othersshuffle顺序仍然随机。第二项策略是学习率衰减。SGD在固定学习率下后期会在最优解附近来回震荡无法精细收敛。使用余弦退火可以自动降低后期学习率让损失进一步下降scheduler torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max10)每个epoch结束后调用scheduler.step()学习率会从0.01平滑下降至接近0。在MNIST任务上这一改动通常能稳定地把准确率从98.7%推到99.1%附近。第三项策略是连续多次训练取最高分。99.12%很可能来自多个随机种子下最好的一次结果。这里要区分“平均准确率”和“最佳准确率”前者更能反映模型真实水平后者用于对标论文数字。建议做法是跑3个不同种子每次10个epoch保留测试集准确率最高的那个权重。4.2 用测试集正确评估model.eval()与no_grad缺一不可评估阶段常见错误是在训练模式下跑测试集这会让权重中的Dropout或BatchNorm行为不一致。LeNet-5没有这两类层但保持正确模式仍然是好习惯。def evaluate(model, test_loader): model.eval() correct 0 total 0 with torch.no_grad(): for images, labels in test_loader: images, labels images.to(device), labels.to(device) outputs model(images) _, predicted torch.max(outputs, 1) total labels.size(0) correct (predicted labels).sum().item() acc 100.0 * correct / total print(fTest Accuracy: {acc:.2f}%) return acc evaluate(model, test_loader)代码逻辑说明model.eval()切换推理模式torch.no_grad()关闭梯度计算图二者共同降低显存占用并保证输出稳定。torch.max(outputs, 1)在维度1上取最大得分对应的索引即预测类别。参数说明如果想复现99.12%这个数字必须记录correct / total的精确值而不是只看打印的两位小数1万张测试集里85张和88张错误分别对应99.12%与99.15%差异很细微。4.3 训练过程的真实进度参考下表是一次典型训练中可能观察到的进度。这个表格的意义在于帮新手建立“正常收敛”的体感如果偏差过大通常说明前面某个环节出了问题。Epoch训练损失测试准确率10.231097.82%30.052198.64%50.030499.05%70.014299.08%100.006399.12%如果第1轮准确率低于90%优先检查标准化参数是否正确如果最终准确率在99%附近波动但始终上不去可以调整T_max为epoch数或者把batch_size降到32。4.4 一个小众但有效的操作不用权重衰减也能稳定MNIST样本量足够大LeNet-5参数量又少约6万通常不需要weight_decay。添加L2正则反而可能小幅降低准确率。这与CIFAR-10等更复杂数据集的经验不同不要盲目套用迁移学习中的配置。5. LeNet-5推理部署与三类常见坑别让模型死在eval阶段5.1 把训练好的模型接到一张真实图片上训练与推理最大的差异是推理时只有一张图且图片可能来自任意来源预处理必须与训练时完全一致。from PIL import Image def predict_image(image_path, model): transform transforms.Compose([ transforms.Resize((28, 28)), transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,)) ]) image Image.open(image_path).convert(L) tensor transform(image).unsqueeze(0).to(device) model.eval() with torch.no_grad(): logits model(tensor) pred torch.argmax(logits, dim1).item() return pred print(predict_image(test_digit.png, model))逻辑说明convert(L)把RGB或RGBA图片转成灰度如果跳过这一步PyTorch会因通道数不匹配直接报错。unsqueeze(0)把形状从[1, 28, 28]扩成[1, 1, 28, 28]补上batch维度。5.2 坑一输入尺寸与原始MNIST不一致很多自定义图片不是正方形的28×28拍出来的手写数字可能是24×24或32×32。LeNet-5对输入尺寸没有自适应能力全连接层维度是写死的。解决方法是先用Resize((28, 28))统一或者继续沿用2.4节的做法在改输入尺寸后重新打印net(dummy).shape。正确做法不是去算C3输出尺寸而是直接写一个快速验证for size in [28, 32, 64]: try: test_net LeNet5() out test_net(torch.randn(1, 1, size, size)) print(f{size}x{size} - {out.shape}) except Exception as e: print(f{size}x{size} - 维度错误: {e})5.3 坑二训练时用了GPU推理时只有CPU权重在GPU上训练后保存的是CUDA张量。加载时会报RuntimeError: Attempting to deserialize object on a CUDA device。解决方法是保存时使用model.state_dict()加载时用map_location指定设备。weights torch.load(lenet5_best.pth, map_locationcpu) model.load_state_dict(weights)参数说明map_locationcpu让所有张量强制迁移到CPU是跨设备推理最省事的方案。如果你的推理机有GPU可以改成map_locationcuda:0。5.4 坑三真正投入使用前忘记验证整个pipeline训练准确率99.12%只能证明模型在MNIST测试集上有效不能证明你的图像分类系统本身就绪。建议写一个冒烟测试准备5张真实图片包含不同背景、不同粗细的笔迹逐一运行predict_image人工核对输出。这一步能暴露从“训练代码”到“可用系统”之间所有遗漏的环节。提示如果推理结果不稳定先检查预处理再检查模型模式最后才怀疑权重损坏。这个顺序覆盖了绝大多数LeNet-5部署失败案例。本文还有配套的精品资源点击获取 PyTorch实现LeNet-5：从网络结构到99.12%准确率实战指南
简介这套基于深度学习LeNet-5模型的图像分类系统源码面向计算机相关专业的本科生毕业设计与课程设计人群适合需要快速搭建图像识别项目、理解经典卷积神经网络训练流程的学习者。项目用Python实现完整流程包含数据清洗与预处理、模型定义和训练、测试评估等模块最终准确率99.12%可作为课程报告或毕设演示的可靠蓝本。资源共27个文件压缩包约11.53MB。其中10个.py源码文件覆盖数据读取、模型构建、训练测试及辅助分析等功能2个.pth文件为训练好的模型参数便于直接加载复现4个.gz为MNIST原始数据集另有文本说明、项目配置等文件整体结构清晰便于按需查看。目前已有102人学习下载适合希望系统参考LeNet-5实际工程实现的初学者。读者可从中获得一套可运行的图像分类项目包括模型结构定义、梯度训练流程、数据增强与预处理技巧、测试脚本及使用说明。通过阅读源码还能掌握深度学习框架下从数据准备到模型部署的常见写法为后续改进或扩展其他分类任务打下基础。1. 从一份带准确率的项目压缩包说起LeNet-5为什么到今天仍值得跑第一次看到“基于深度学习LeNet-5模型的图像分类系统python源码准确率99.12%.zip”这样的标题大多数人的第一反应可能是“2026年了还有人用1998年的网络做分类”。这个反应本身没有错但恰恰忽略了一个事实LeNet-5在MNIST这类灰度小图像任务上配合正确的训练策略依然能以极小的参数量获得99%以上的准确率。换句话说这份压缩包卖的不是“新”而是“经典结构正确的训练方法”的组合。这个标题实际指向一个很具体的需求用Python实现一个能直接运行的图像分类系统输入手写数字图片输出类别并且准确率要能够对标论文级别的99.12%。它解决的典型场景包括课程设计、算法验证、简历项目以及刚入门深度学习时“第一个能跑通并看到高准确率”的正反馈。适合的人群是熟悉Python基础、但还没有完整训练过一个CNN模型的开发者。接下来要做的是从网络结构、数据处理、训练策略到推理部署把这个“99.12%”拆成一串可复现的参数和代码。2. LeNet-5网络结构拆解从C1到F6的参数表与PyTorch实现2.1 经典结构在MNIST上的两种改法LeNet-5原论文面向的是32×32灰度输入结构依次为卷积层C1、池化层S2、卷积层C3、池化层S4、卷积层C5、全连接层F6和输出层。原始C5是一个卷积分支输出的是一维特征向量在MNIST的28×28输入下很多实现会直接把C5改写成全连接层或者给第一层卷积加padding。两种改法都不影响“LeNet-5”这个名称的成立但直接影响后续全连接层的维度计算。第一种改法输入28×28第一层卷积不加padding5×5卷积核输出24×24池化后12×12第二种改法第一层卷积加padding2输出28×28池化后14×14。这两种计算路径在PyTorch中都很常见区别只在于你对原始架构还原到什么程度。下面采用padding2的方案因为它能让网络在MNIST上保留更多空间信息收敛更快这也是许多高准确率复现的默认选择。2.2 每一层在做什么参数表与计算逻辑以“1个灰度通道输入 → 6个5×5卷积核 → 平均池化 → 16个5×5卷积核 → 平均池化 → 展平 → 120 → 84 → 10”为骨干各层输出尺寸与参数量如下表层名操作卷积核/池化输出尺寸参数量C1卷积5×5, 6个28×28×6156S2平均池化2×2, stride 214×14×60C3卷积5×5, 16个10×10×162416S4平均池化2×2, stride 25×5×160C5全连接400→12012048120F6全连接120→848410164输出全连接84→1010850注意C3的参数量是2416而不是简单的16×25。原因是C3的每个卷积核都作用在S2的全部6个通道上所以单个卷积核的权重是6×5×5150加上1个偏置再乘以16个卷积核得到(1501)×162416。这个细节在阅读源码时容易困惑也是调试网络结构时最常被忽视的地方。2.3 用PyTorch实现一个可训练的LeNet-5下面这段代码定义了完整的网络输入为1通道灰度图输出为10类得分import torch.nn as nn class LeNet5(nn.Module): def __init__(self, num_classes10): super().__init__() self.features nn.Sequential( # C1: 1通道 - 6通道, padding2 保持28x28 nn.Conv2d(1, 6, kernel_size5, padding2), nn.ReLU(), # S2: 平均池化, 输出14x14 nn.AvgPool2d(kernel_size2, stride2), # C3: 6通道 - 16通道, 输出10x10 nn.Conv2d(6, 16, kernel_size5), nn.ReLU(), # S4: 输出5x5 nn.AvgPool2d(kernel_size2, stride2), ) self.classifier nn.Sequential( nn.Flatten(), # C5与F6: 16*5*5400 - 120 - 84 nn.Linear(16 * 5 * 5, 120), nn.ReLU(), nn.Linear(120, 84), nn.ReLU(), nn.Linear(84, num_classes) ) def forward(self, x): return self.classifier(self.features(x))代码逻辑说明features负责从原始像素中提取局部纹理和边缘组合classifier负责把空间特征映射到类别概率。nn.AvgPool2d是LeNet原始使用的平均池化与现在图像分类中更主流的nn.MaxPool2d不同平均池化对微小位移更平滑在MNIST上两者差距很小但早期复现项目里多为平均池化。参数说明padding2是保证28×28输入下C1输出仍然为28×28的关键去掉它会让后续nn.Linear(16 * 5 * 5, 120)直接报维度错误。如果你希望严格还原32×32输入可以把网络第一层去掉padding但需要把输入图像resize到32×32同时修改全连接层输入维度为16×5×5400在32×32输入加padding0时C3输出尺寸会不同务必用下面的验证方法确认。2.4 一个必须养成的习惯实例化后立即验证输出形状不要凭直觉去猜维度用一行代码确认网络输出import torch net LeNet5() dummy torch.randn(1, 1, 28, 28) print(net(dummy).shape) # torch.Size([1, 10])这一步能过滤掉绝大多数传入尺寸错误。如果输出不是[1, 10]优先检查卷积padding和全连接层输入维度是否与特征图展平后的长度一致。3. MNIST数据加载、标准化与训练循环跑通99%准确率的最小代码3.1 数据集的获取与预处理为什么要用这两个标准化参数MNIST是LeNet-5最经典的实验场PyTorch的torchvision.datasets.MNIST可以直接下载。训练集6万张、测试集1万张每张是28×28灰度图。预处理只有两步转成Tensor、标准化。import torch from torch.utils.data import DataLoader from torchvision import datasets, transforms transform transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,)) ]) train_dataset datasets.MNIST(root./data, trainTrue, downloadTrue, transformtransform) test_dataset datasets.MNIST(root./data, trainFalse, downloadTrue, transformtransform) train_loader DataLoader(train_dataset, batch_size64, shuffleTrue) test_loader DataLoader(test_dataset, batch_size1000, shuffleFalse)逻辑说明归一化让像素值从[0,1]映射到近似标准正态分布可以加速收敛。0.1307是MNIST训练集的全局均值0.3081是全局标准差这两个数值是torchvision文档和各类复现项目长期验证的固定值不必自己重算。参数说明shuffleTrue只用于训练集它打断样本顺序避免模型学到数据排列中的假规律测试集不需要打乱。batch_size64对于LeNet-5来说是常规选择显存占用很低CPU上也能流畅训练如果你用的是GPU可以提高到128或256准确率不会受到明显影响。3.2 训练循环SGD与交叉熵的组合下面是一个精简但完整的训练流程保留了进度输出与模型保存import torch.optim as optim import torch.nn as nn device torch.device(cuda if torch.cuda.is_available() else cpu) model LeNet5().to(device) criterion nn.CrossEntropyLoss() optimizer optim.SGD(model.parameters(), lr0.01, momentum0.9) def train_one_epoch(epoch): model.train() total_loss 0 for images, labels in train_loader: images, labels images.to(device), labels.to(device) optimizer.zero_grad() outputs model(images) loss criterion(outputs, labels) loss.backward() optimizer.step() total_loss loss.item() avg_loss total_loss / len(train_loader) print(fEpoch {epoch} | Loss: {avg_loss:.4f}) return avg_loss for epoch in range(1, 11): train_one_epoch(epoch) torch.save(model.state_dict(), flenet5_epoch{epoch}.pth)代码逻辑说明model.train()开启训练模式optimizer.zero_grad()清空梯度loss.backward()反向传播optimizer.step()更新权重。顺序不能颠倒漏掉zero_grad()会导致梯度累积表现为损失曲线抖动甚至发散。参数说明lr0.01配合momentum0.9是MNIST上最稳妥的起点。动量项让参数更新在梯度方向变化较小时保持惯性能有效穿过局部极小值。对比AdamSGD在这个任务上更容易稳定收敛到99%以上的区域但如果你用的是更深的网络或非MNIST数据集Adam可能更好调。损失函数用CrossEntropyLoss它内部已经包含Softmax计算不需要在网络输出层再手动加Softmax。3.3 训练参数速查表下表是复现99%以上准确率的一组推荐参数直接照抄即可起步。参数推荐值说明优化器SGDmomentum0.9比Adam更贴合经典复现学习率0.01过大容易震荡过小收敛慢Batch Size64CPU友好稳定性和速度的折中Epoch10第5轮左右可到99%10轮彻底收敛数据预处理Mean0.1307, Std0.3081固定值不要自行计算池化方式AvgPool2d还原LeNet原始设计也可替换为MaxPool2d如果训练到第10轮准确率卡在98%左右大概率不是网络代码问题而是随机种子、数据顺序或学习率策略的影响。这正是下面一章要展开的内容。4. 从97%到99.12%LeNet-5的训练策略、超参与准确率验证4.1 为什么同样的网络有人99.12%有人98.3%随机种子、shuffle顺序、学习率是否衰减、池化层选平均还是最大这些因素叠加在一起会在MNIST上产生0.5%到1%的准确率波动。99.12%并不是一个靠运气才能碰到的天花板而是几项关键策略同时起作用的自然结果。第一项策略是固定随机种子。PyTorch中数据加载与权重初始化都依赖随机数不固定种子的话每次运行结果会有细微差异。在训练脚本开头加入import random import numpy as np import torch def set_seed(seed42): random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed_all(seed) set_seed(42)逻辑说明torch.manual_seed控制权重初始化和CUDA相关随机数random.seed与np.random.seed影响数据集shuffle。三者缺一不可只固定torch而忽略othersshuffle顺序仍然随机。第二项策略是学习率衰减。SGD在固定学习率下后期会在最优解附近来回震荡无法精细收敛。使用余弦退火可以自动降低后期学习率让损失进一步下降scheduler torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max10)每个epoch结束后调用scheduler.step()学习率会从0.01平滑下降至接近0。在MNIST任务上这一改动通常能稳定地把准确率从98.7%推到99.1%附近。第三项策略是连续多次训练取最高分。99.12%很可能来自多个随机种子下最好的一次结果。这里要区分“平均准确率”和“最佳准确率”前者更能反映模型真实水平后者用于对标论文数字。建议做法是跑3个不同种子每次10个epoch保留测试集准确率最高的那个权重。4.2 用测试集正确评估model.eval()与no_grad缺一不可评估阶段常见错误是在训练模式下跑测试集这会让权重中的Dropout或BatchNorm行为不一致。LeNet-5没有这两类层但保持正确模式仍然是好习惯。def evaluate(model, test_loader): model.eval() correct 0 total 0 with torch.no_grad(): for images, labels in test_loader: images, labels images.to(device), labels.to(device) outputs model(images) _, predicted torch.max(outputs, 1) total labels.size(0) correct (predicted labels).sum().item() acc 100.0 * correct / total print(fTest Accuracy: {acc:.2f}%) return acc evaluate(model, test_loader)代码逻辑说明model.eval()切换推理模式torch.no_grad()关闭梯度计算图二者共同降低显存占用并保证输出稳定。torch.max(outputs, 1)在维度1上取最大得分对应的索引即预测类别。参数说明如果想复现99.12%这个数字必须记录correct / total的精确值而不是只看打印的两位小数1万张测试集里85张和88张错误分别对应99.12%与99.15%差异很细微。4.3 训练过程的真实进度参考下表是一次典型训练中可能观察到的进度。这个表格的意义在于帮新手建立“正常收敛”的体感如果偏差过大通常说明前面某个环节出了问题。Epoch训练损失测试准确率10.231097.82%30.052198.64%50.030499.05%70.014299.08%100.006399.12%如果第1轮准确率低于90%优先检查标准化参数是否正确如果最终准确率在99%附近波动但始终上不去可以调整T_max为epoch数或者把batch_size降到32。4.4 一个小众但有效的操作不用权重衰减也能稳定MNIST样本量足够大LeNet-5参数量又少约6万通常不需要weight_decay。添加L2正则反而可能小幅降低准确率。这与CIFAR-10等更复杂数据集的经验不同不要盲目套用迁移学习中的配置。5. LeNet-5推理部署与三类常见坑别让模型死在eval阶段5.1 把训练好的模型接到一张真实图片上训练与推理最大的差异是推理时只有一张图且图片可能来自任意来源预处理必须与训练时完全一致。from PIL import Image def predict_image(image_path, model): transform transforms.Compose([ transforms.Resize((28, 28)), transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,)) ]) image Image.open(image_path).convert(L) tensor transform(image).unsqueeze(0).to(device) model.eval() with torch.no_grad(): logits model(tensor) pred torch.argmax(logits, dim1).item() return pred print(predict_image(test_digit.png, model))逻辑说明convert(L)把RGB或RGBA图片转成灰度如果跳过这一步PyTorch会因通道数不匹配直接报错。unsqueeze(0)把形状从[1, 28, 28]扩成[1, 1, 28, 28]补上batch维度。5.2 坑一输入尺寸与原始MNIST不一致很多自定义图片不是正方形的28×28拍出来的手写数字可能是24×24或32×32。LeNet-5对输入尺寸没有自适应能力全连接层维度是写死的。解决方法是先用Resize((28, 28))统一或者继续沿用2.4节的做法在改输入尺寸后重新打印net(dummy).shape。正确做法不是去算C3输出尺寸而是直接写一个快速验证for size in [28, 32, 64]: try: test_net LeNet5() out test_net(torch.randn(1, 1, size, size)) print(f{size}x{size} - {out.shape}) except Exception as e: print(f{size}x{size} - 维度错误: {e})5.3 坑二训练时用了GPU推理时只有CPU权重在GPU上训练后保存的是CUDA张量。加载时会报RuntimeError: Attempting to deserialize object on a CUDA device。解决方法是保存时使用model.state_dict()加载时用map_location指定设备。weights torch.load(lenet5_best.pth, map_locationcpu) model.load_state_dict(weights)参数说明map_locationcpu让所有张量强制迁移到CPU是跨设备推理最省事的方案。如果你的推理机有GPU可以改成map_locationcuda:0。5.4 坑三真正投入使用前忘记验证整个pipeline训练准确率99.12%只能证明模型在MNIST测试集上有效不能证明你的图像分类系统本身就绪。建议写一个冒烟测试准备5张真实图片包含不同背景、不同粗细的笔迹逐一运行predict_image人工核对输出。这一步能暴露从“训练代码”到“可用系统”之间所有遗漏的环节。提示如果推理结果不稳定先检查预处理再检查模型模式最后才怀疑权重损坏。这个顺序覆盖了绝大多数LeNet-5部署失败案例。本文还有配套的精品资源点击获取